fix unittest

skypilot-org · Mar 25, 2024 · 25ea7b3 · 25ea7b3
1 parent 245848e
commit 25ea7b3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/sky/utils/controller_utils.py b/sky/utils/controller_utils.py
@@ -264,8 +264,8 @@ def get_controller_resources(
     controller_type: str,
     controller_name: str,
     controller_resources_config: Dict[str, Any],
-    task_resources: Iterable[resources.Resources],
-) -> Set[resources.Resources]:
+    task_resources: Iterable['resources.Resources'],
+) -> Set['resources.Resources']:
     """Read the skypilot config and setup the controller resources.
 
     Returns: