def METHOD_NAME(self, target : Target, args : Tuple[Argument, ...], kwargs : Dict[str, Any]):
    python_ret_type = None
    assert isinstance(target, str)
    submod = self.fetch_attr(target)
    if self.annotate_modules and hasattr(submod.__class__, '__name__'):
        classname = submod.__class__.__name__
        if getattr(torch.nn, classname, None) == submod.__class__:
            python_ret_type = self._extract_python_return_type(submod.forward)
    return_proxy = super().METHOD_NAME(target, args, kwargs)
    return_proxy.node.type = return_proxy.node.type if return_proxy.node.type else python_ret_type
    return return_proxy