fix dep version locking

autonomio · Apr 21, 2024 · c3df276 · c3df276
1 parent 9ef50bb
commit c3df276
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -36,8 +36,8 @@ dependencies = [
     "pandas",
     "requests",
     "scikit-learn",
-    "statsmodels>=0.11.0",
-    "tensorflow==2.14.1",
+    "statsmodels = ">=0.11.0",
+    "tensorflow = "==2.14.1",
     "tqdm",
     "wrangle",
 ]

diff --git a/tests/commands/test_scan.py b/tests/commands/test_scan.py
@@ -20,7 +20,6 @@ def test_scan():
          'dropout': (.05, .35, .1),
          'epochs': [50]}
 
-    @tf.function
     def iris_model(x_train, y_train, x_val, y_val, params):
 
         model = Sequential()