address review comment, type hint

ollmer · May 13, 2024 · 3d28971 · 3d28971
1 parent ffba574
commit 3d28971
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/sweagent/environment/swe_env.py b/sweagent/environment/swe_env.py
@@ -462,9 +462,10 @@ def reset_container(self) -> None:
         self.container_obj = None
         self._reset_container()
 
-    def _init_container(self, cached_image=None) -> None:
+    def _init_container(self, cached_image: Optional[str] = None) -> None:
         """
-        Handles container initialization. Defines container name and creates it
+        Handles container initialization. Defines container name and creates it.
+        If cached_image is provided, it will use that image name instead of the default.
         """
         image_name = self.image_name
         if cached_image is not None: