Merge pull request #2136 from devharsh/patch-1

Update model-stealing-demo.ipynb
Trusted-AI · May 22, 2023 · 50b2a8b · 50b2a8b
2 parents 6029b12 + aee4c25
commit 50b2a8b
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/notebooks/model-stealing-demo.ipynb b/notebooks/model-stealing-demo.ipynb
@@ -35,10 +35,9 @@
     "from keras.models import Sequential\n",
     "from keras.layers import Dense, Flatten, Conv2D, MaxPooling2D, Flatten, InputLayer, Reshape\n",
     "\n",
-    "from art.classifiers import KerasClassifier\n",
+    "from art.estimators.classification import KerasClassifier\n",
     "\n",
-    "if tf.executing_eagerly():\n",
-    "    tf.python.framework.ops.disable_eager_execution()"
+    "tf.compat.v1.disable_eager_execution()"
    ]
   },
   {