Fix issue handling function with decorator

Differential Revision: D82132556

Pull Request resolved: #1111

Author: nghuiqin

torchx/specs/finder.py

@@ -287,7 +287,9 @@ def _get_path_to_function_decl(
         my_component defined in some_file.py, imported in other_file.py
         and the component is invoked as other_file.py:my_component
         """
-        path_to_function_decl = inspect.getabsfile(function)
+        # Unwrap decorated functions to get the original function
+        unwrapped_function = inspect.unwrap(function)
+        path_to_function_decl = inspect.getabsfile(unwrapped_function)
         if path_to_function_decl is None or not os.path.isfile(path_to_function_decl):
             return self._filepath
         return path_to_function_decl