From dcabe89608b60b84036504ace22655647be54bd9 Mon Sep 17 00:00:00 2001
From: NicolaCourtier <45851982+NicolaCourtier@users.noreply.github.com>
Date: Mon, 5 Aug 2024 11:15:19 +0100
Subject: [PATCH] Update check for ECM init_soc

---
 pybop/models/base_model.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/pybop/models/base_model.py b/pybop/models/base_model.py
index 36907eff..05dff1c4 100644
--- 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