Commit fd0c85c0 authored by VIGNESHinZONE's avatar VIGNESHinZONE
Browse files

fix try catch

parent 24acca18

deepchem/models/jax_models/jax_model.py

+4 −7
Original line number Diff line number Diff line
@@ -14,13 +14,10 @@ from typing import Any, Callable, Iterable, List, Optional, Tuple, Union 
from deepchem.utils.typing import LossFn



# JAX depend

try:

import jax.numpy as jnp

import jax

import haiku as hk

import optax

except:

  raise ImportError("This class requires Jax, haiku and optax to be installed.")



import warnings

CRA Git | Maintained and supported by SUSTech CRA and CCSE