Update check for ECM init_soc

pybop-team · Aug 5, 2024 · dcabe89 · dcabe89
1 parent baca0d7
commit dcabe89
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/pybop/models/base_model.py b/pybop/models/base_model.py
@@ -517,7 +517,9 @@ def predict(
             inputs = self.parameters.verify(inputs)
             parameter_set.update(inputs)
 
-        if init_soc is not None and "Initial SoC" in parameter_set.keys():
+        if init_soc is not None and isinstance(
+            self.pybamm_model, pybamm.equivalent_circuit.Thevenin
+        ):
             parameter_set["Initial SoC"] = init_soc
             init_soc = None