From 12e083ebc44a55512c4d500df6cf0f370cab73bc Mon Sep 17 00:00:00 2001
From: Will Handley <wh260@cam.ac.uk>
Date: Thu, 18 Apr 2024 16:21:59 +0100
Subject: [PATCH] Updated dependencies

---
 pyproject.toml | 3 ++-
 stemu/emu.py   | 2 +-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index d8cf89d..6984731 100644
--- 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
index a95afdf..d16f780 100644
--- 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")]
         )