I’m using vs-code with pylance for intellisense, and my op signatures are coming back as

(function) func_name: SolidDefinition

due to the decorator. I had a look and it seemed like modifying the type definitions in op_decorator.py in the following way:

# existing
@overload
def op(name: Callable[..., Any]) -> SolidDefinition:
    ...

# new
_TFunc = TypeVar("_TFunc", bound=Callable[..., Any])


@overload
def op(name: _TFunc) -> Union[_TFunc, SolidDefinition]:
     ...

fixes it, but wanted to check here whether you guys had already considered this / excluded it as a fix for any reason

cc @sean

Hi johnny, just letting you know that I’m looking into this for next week’s release.

Ah, that’s great. Thanks sean