pyright complained

Author: pmcurtin

examples/slurm_poc.py

@@ -11,7 +11,6 @@
 def test_launch():
     result = torchrunx.launch(
         func=simple_matmul,
-        func_kwargs={},
         hostnames=torchrunx.slurm_hosts(),
         workers_per_host=torchrunx.slurm_workers(),
     )
@@ -22,7 +21,7 @@ def test_launch():
     print("PASS")
 
 
-def simple_matmul():
+def simple_matmul(test):
     rank = int(os.environ["RANK"])
     local_rank = int(os.environ["LOCAL_RANK"])
     device = torch.device(local_rank) if torch.cuda.is_available() else torch.device("cpu")
@@ -36,7 +35,7 @@ def simple_matmul():
 
     i = torch.rand((500, 100), device=device)  # batch, dim
     o = torch.matmul(i, w)
-
+    print(test)
     dist.all_reduce(o, op=dist.ReduceOp.SUM)
     print(i)
     return o.detach().cpu()

src/torchrunx/launcher.py

@@ -101,7 +101,7 @@ class Launcher:
     def run(
         self,
         func: Callable,
-        func_args: tuple[Any] = (),
+        func_args: tuple[Any] = tuple(),
         func_kwargs: dict[str, Any] = {},
     ) -> dict[int, Any]:
         """
@@ -258,7 +258,7 @@ def run(
 
 def launch(
     func: Callable,
-    func_args: tuple[Any] = (),
+    func_args: tuple[Any] = tuple(),
     func_kwargs: dict[str, Any] = {},
     hostnames: list[str] = ["localhost"],
     workers_per_host: int | list[int] = 1,