Adding new imports so code will run

nasa · Oct 13, 2023 · 9a514be · 9a514be
1 parent 022247e
commit 9a514be
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/examples/04_New Models.ipynb b/examples/04_New Models.ipynb
@@ -1427,7 +1427,8 @@
     "import matplotlib.pyplot as plt\n",
     "import pickle\n",
     "import numpy as np\n",
-    "from progpy.models import BatteryElectroChemEOD"
+    "from progpy.models import BatteryElectroChemEOD\n",
+    "from progpy.loading import Piecewise"
    ]
   },
   {
@@ -1527,7 +1528,7 @@
    "source": [
     "# Variable (piecewise) future loading scheme\n",
     "future_loading = Piecewise(\n",
-    "        m.InputContainer,\n",
+    "        batt.InputContainer,\n",
     "        [600, 1000, 1500, 3000],\n",
     "        {'i': [3, 2, 1.5, 4]})"
    ]
@@ -1603,7 +1604,7 @@
     "                  np.array_equal(voltage_orig, voltage_json_1) and \\\n",
     "                  np.array_equal(voltage_orig, voltage_json_2)\n",
     "\n",
-    "print(f\"The simulated results from the original and serialized models are {'identical. This means that our serialization works!' if are_arrays_same else 'not identical. This means that our serialization does not work.'}!\")"
+    "print(f\"The simulated results from the original and serialized models are {'identical. This means that our serialization works!' if are_arrays_same else 'not identical. This means that our serialization does not work.'}\")"
    ]
   },
   {