Update imports

Signed-off-by: Beat Buesser <[email protected]>

Author: Beat Buesser

art/estimators/classification/__init__.py

@@ -14,6 +14,7 @@
 from art.estimators.classification.ensemble import EnsembleClassifier
 from art.estimators.classification.GPy import GPyGaussianProcessClassifier
 from art.estimators.classification.keras import KerasClassifier
+from art.estimators.classification.jax import JaxClassifier
 from art.estimators.classification.lightgbm import LightGBMClassifier
 from art.estimators.classification.mxnet import MXClassifier
 from art.estimators.classification.pytorch import PyTorchClassifier

art/utils.py

@@ -77,6 +77,7 @@
     from art.estimators.classification.ensemble import EnsembleClassifier
     from art.estimators.classification.GPy import GPyGaussianProcessClassifier
     from art.estimators.classification.keras import KerasClassifier
+    from art.estimators.classification.jax import JaxClassifier
     from art.estimators.classification.lightgbm import LightGBMClassifier
     from art.estimators.classification.mxnet import MXClassifier
     from art.estimators.classification.pytorch import PyTorchClassifier
@@ -111,6 +112,7 @@
         EnsembleClassifier,
         GPyGaussianProcessClassifier,
         KerasClassifier,
+        JaxClassifier,
         MXClassifier,
         PyTorchClassifier,
         ScikitlearnLogisticRegression,
@@ -162,6 +164,7 @@
         EnsembleClassifier,
         GPyGaussianProcessClassifier,
         KerasClassifier,
+        JaxClassifier,
         LightGBMClassifier,
         MXClassifier,
         PyTorchClassifier,