[Core] Static type checking fails for ray.remote when called with options

[birgerbr]

What happened + What you expected to happen

Static type checking with mypy and PyCharm fails when ray.remote is used with options (ray.remote(num_cpus=2)).

Running mypy on this code:

import ray


@ray.remote(num_cpus=2)
def f():
    pass

gives the following error message:

error: No overload variant of "remote" matches argument type "int"
note: Possible overload variants:
note:     def [R] remote(function: Callable[[], R]) -> RemoteFunction[R, None, None, None, None, None, None, None, None, None, None]
note:     def [T0, R] remote(function: Callable[[T0], R]) -> RemoteFunction[R, T0, None, None, None, None, None, None, None, None, None]
note:     def [T0, T1, R] remote(function: Callable[[T0, T1], R]) -> RemoteFunction[R, T0, T1, None, None, None, None, None, None, None, None]
note:     def [T0, T1, T2, R] remote(function: Callable[[T0, T1, T2], R]) -> RemoteFunction[R, T0, T1, T2, None, None, None, None, None, None, None]
note:     def [T0, T1, T2, T3, R] remote(function: Callable[[T0, T1, T2, T3], R]) -> RemoteFunction[R, T0, T1, T2, T3, None, None, None, None, None, None]
note:     def [T0, T1, T2, T3, T4, R] remote(function: Callable[[T0, T1, T2, T3, T4], R]) -> RemoteFunction[R, T0, T1, T2, T3, T4, None, None, None, None, None]
note:     def [T0, T1, T2, T3, T4, T5, R] remote(function: Callable[[T0, T1, T2, T3, T4, T5], R]) -> RemoteFunction[R, T0, T1, T2, T3, T4, T5, None, None, None, None]
note:     def [T0, T1, T2, T3, T4, T5, T6, R] remote(function: Callable[[T0, T1, T2, T3, T4, T5, T6], R]) -> RemoteFunction[R, T0, T1, T2, T3, T4, T5, T6, None, None, None]
note:     def [T0, T1, T2, T3, T4, T5, T6, T7, R] remote(function: Callable[[T0, T1, T2, T3, T4, T5, T6, T7], R]) -> RemoteFunction[R, T0, T1, T2, T3, T4, T5, T6, T7, None, None]
note:     def [T0, T1, T2, T3, T4, T5, T6, T7, T8, R] remote(function: Callable[[T0, T1, T2, T3, T4, T5, T6, T7, T8], R]) -> RemoteFunction[R, T0, T1, T2, T3, T4, T5, T6, T7, T8, None]
note:     def [T0, T1, T2, T3, T4, T5, T6, T7, T8, T9, R] remote(function: Callable[[T0, T1, T2, T3, T4, T5, T6, T7, T8, T9], R]) -> RemoteFunction[R, T0, T1, T2, T3, T4, T5, T6, T7, T8, T9]
note:     def remote(t: type) -> Any
Found 1 error in 1 file (checked 1 source file)

This was not an issue with the previous ray version we were using (1.10.0).

Versions / Dependencies

Ray: 1.13.0
Python: 3.8.10
OS: Ubuntu 20.04.4 LTS

Reproduction script

test_ray_remote.py:

import ray


@ray.remote(num_cpus=2)
def f():
    pass

mypy test_ray_remote.py`

Issue Severity

High: It blocks me from completing my task.

[DmitriGekhtman]

cc @iycheng

[pcmoritz]

@tiangolo Can you have a look at this? :)

[simenandresen]

@DmitriGekhtman @pcmoritz @tiangolo do you have any idea if we can expect a fix for this soon?

[tiangolo]

Yep! Here's the PR with the fix 🤓 #25999