From 49f855c32072cb2df4a74e1be60a3834776b366a Mon Sep 17 00:00:00 2001
From: Chris McCarthy <chris.mccarthy@methods.co.uk>
Date: Thu, 10 Aug 2023 13:33:32 +0100
Subject: [PATCH] #1706 - Synced with Dev

---
 src/primaite/simulator/core.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/primaite/simulator/core.py b/src/primaite/simulator/core.py
index 8b771cd7..7a183588 100644
--- a/src/primaite/simulator/core.py
+++ b/src/primaite/simulator/core.py
@@ -177,7 +177,7 @@ class SimComponent(BaseModel):
         """
         pass
 
-    def reset_component_for_episode(self):
+    def reset_component_for_episode(self, episode: int):
         """
         Reset this component to its original state for a new episode.