fix keras tests

ucsd-ets · Sep 29, 2024 · 5900a6b · 5900a6b
1 parent 6b4a197
commit 5900a6b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/images/scipy-ml-notebook/workflow_tests/test_keras.py b/images/scipy-ml-notebook/workflow_tests/test_keras.py
@@ -1,6 +1,6 @@
 import numpy as np
-from keras.models import Sequential
-from keras.layers import Dense
+from tf_keras.models import Sequential
+from tf_keras.layers import Dense
 import pytest
 
 @pytest.fixture