From 4d4857bc92a0a93594f3a9a47fbedaaab5a8c4a2 Mon Sep 17 00:00:00 2001
From: NicolaCourtier <45851982+NicolaCourtier@users.noreply.github.com>
Date: Mon, 5 Aug 2024 15:27:32 +0100
Subject: [PATCH] Fix predict without pybamm test

---
 pybop/models/base_model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pybop/models/base_model.py b/pybop/models/base_model.py
index 08339527..b02cc43b 100644
--- a/pybop/models/base_model.py
+++ b/pybop/models/base_model.py
@@ -505,7 +505,7 @@ def predict(
             if PyBaMM models are not supported by the current simulation method.
 
         """
-        if self._unprocessed_model is None:
+        if self.pybamm_model is None:
             raise ValueError(
                 "The predict method currently only supports PyBaMM models."
             )