Merge pull request #182 from lymichelle21/fix-tests

Fix test failures

Author: teubert

pyproject.toml

@@ -57,8 +57,14 @@ classifiers = [
 
 [project.optional-dependencies]
 datadriven = [
-    "tensorflow; platform_system!='Darwin' or platform_machine!='arm64'",
-    "tensorflow-macos; platform_system=='Darwin' and platform_machine=='arm64'"]
+    "tensorflow"
+]
+
+test = [
+    "notebook",
+    "testbook",
+    "requests"
+]
 
 docs = [
     "sphinx",

src/progpy/data_models/lstm_model.py

@@ -585,7 +585,7 @@ def from_data(cls, inputs, outputs, event_states=None, t_met=None, **kwargs):
             output_data.append(t_all)
 
         model = keras.Model(inputs, outputs)
-        model.compile(optimizer="rmsprop", loss="mse", metrics=[["mae"]]*len(outputs))
+        model.compile(optimizer="rmsprop", loss="mse", metrics=["mae"]*len(outputs))
 
         # Train model
         history = model.fit(

test_requirements.txt

@@ -1654,7 +1654,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from prog_algs.uncertain_data import ScalarData"
+    "from progpy.uncertain_data import ScalarData"
    ]
   },
   {