typing fix

Lightning-AI · Jul 11, 2024 · 52b4b86 · 52b4b86
1 parent 338a6b2
commit 52b4b86
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/litdata/streaming/resolver.py b/src/litdata/streaming/resolver.py
@@ -343,7 +343,7 @@ def _resolve_time_template(path: str) -> str:
 def _execute(
     name: str,
     num_nodes: int,
-    machine: Optional[Machine] = None,
+    machine: Optional["Machine"] = None,
     command: Optional[str] = None,
 ) -> None:
     """Remotely execute the current operator."""