Updated dependencies

handley-lab · Apr 18, 2024 · 12e083e · 12e083e
1 parent c4b9f46
commit 12e083e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -18,7 +18,8 @@ dependencies = [
     'scipy',
     'matplotlib',
     'pandas',
-    'keras'
+    'tensorflow',
+    'scikit-learn',
 ]
 classifiers = [
     "Programming Language :: Python :: 3",

diff --git a/stemu/emu.py b/stemu/emu.py
@@ -75,7 +75,7 @@ def fit(self, X, t, y):
         X, y = stack(X, t, y)
 
         self.model = keras.models.Sequential(
-            [keras.layers.Input(X.shape[-1])]
+            [keras.layers.Input(X.shape[-1:])]
             + default_network
             + [keras.layers.Dense(1, activation="linear")]
         )