Support type checking for launch arguments

[apoorvkh]

No description provided.

[apoorvkh]

Don't think there is a solution here. This is similar to functools.partial, which needs/needed a custom patch in pyright/mypy.

[apoorvkh]

Actually, maybe there are some ways to do this:

python/typeshed#8703
https://peps.python.org/pep-0612/
python/mypy#1484

[apoorvkh]

Consider adding a decorator?

@torchrunx.launch(hostnames=[...])
def my_function(x: int):
   ...

my_function(x=10)  # this command is launched and is also statically type checked

https://github.com/chrisgrimm/better_partial

[apoorvkh]

See typing.ParamSpec

[apoorvkh]

ChatGPT suggestion

from typing import Callable, TypeVar, ParamSpec

P = ParamSpec("P")
R = TypeVar("R")

def call_func(func: Callable[P, R], args: P.args) -> R:
    return func(*args)

# Example usage
def add(a: int, b: int) -> int:
    return a + b

result = call_func(add, (1, 2))  # ✅ Correct usage
# result = call_func(add, (1, 2, 3))  # ❌ Type checker will flag this as an error
# result = call_func(add, ("one", "two"))  # ❌ Type checker will flag this as an error

print(result)  # Output: 3