from typing import Callable


def _method(self: "Foo") -> "Foo":
    return Foo()


def _method_wrapper(
) -> Callable[["Foo"], "Foo"]:
    return _method


class Foo:    
    direct_assignment =  _method
    assign_via_wrapper = _method_wrapper()


def boo(c: Foo) -> None:
    c.direct_assignment()
    c.assign_via_wrapper()
# | ^^^^^^^^^^^^^^^^^^^^^^
# |
# info: Union variant `(Foo, /) -> Foo` is incompatible with this call site
# info: Attempted to call union type `Unknown | ((Foo, /) -> Foo)`
# info: rule `missing-argument` is enabled by default

from pyspark.sql.column import Column
def foo(c: Column) -> Column:
    return c.desc()