Need to add overload functions for allure.step()

[anci3ntr0ck]

I'm submitting a ...

• [x] bug report

What is the current behavior?

VSCode gives an error when using allure.step() as a context:

Adding overload functions for allure.step() should fix this...

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

@overload
def step(title: _TFunc) -> _TFunc: ...
@overload
def step(title: Any) -> StepContext: ...

Please tell us about your environment:

• Allure adaptor: [email protected]

[rafaelcs]

Same here! I'm getting this Pylance warning:

Object of type "(...) -> Any" cannot be used with "with" because it does not implement __enter__Pylance(reportGeneralTypeIssues)

[dc-avasilev]

+1