Fixing failing tests

Signed-off-by: Marc Romeyn <[email protected]>

Author: marcromeyn

nemo_run/lazy.py

@@ -126,7 +126,11 @@ def resolve(self, ctx: Optional["RunContext"] = None) -> Partial:
             if isinstance(fn, LazyTarget):
                 fn = fn.target
 
-        sig = inspect.signature(fn)
+        _fn = fn
+        if hasattr(fn, "__fn_or_cls__"):
+            _fn = fn.__fn_or_cls__
+
+        sig = inspect.signature(_fn)
         param_names = sig.parameters.keys()
 
         dotlist = dictconfig_to_dot_list(