Commit 08e7467e authored by VIGNESHinZONE's avatar VIGNESHinZONE
Browse files

setup fix

parent d865275f

setup.py

+1 −1
Original line number Diff line number Diff line
@@ -11,7 +11,7 @@ else: 


# Environment-specific dependencies.

extras = {

    'jax': ['jax', 'jaxlib', 'dm-haiku=', 'optax'],

    'jax': ['jax', 'jaxlib', 'dm-haiku', 'optax'],

    'torch': ['torch', 'torchvision', 'dgl', 'dgllife'],

    'tensorflow': [

        'tensorflow==2.5.*', 'tensorflow_probability==0.13.*',

CRA Git | Maintained and supported by SUSTech CRA and CCSE