Update pytorch_lightning/utilities/params_tying.py

Co-authored-by: ananthsub <[email protected]>

Author: kaushikb11

pytorch_lightning/utilities/params_tying.py

@@ -48,22 +48,22 @@ def find_shared_parameters(
         return [x for x in tied_parameters.values() if len(x) > 1]
 
 
-def set_shared_parameters(module: "pl.LightningModule", shared_params: list):
+def set_shared_parameters(module: nn.Module, shared_params: list):
     for shared_param in shared_params:
         ref = _get_module_by_path(module, shared_param[0])
         for path in shared_param[1:]:
             _set_module_by_path(module, path, ref)
     return module
 
 
-def _get_module_by_path(module: "pl.LightningModule", path: str):
+def _get_module_by_path(module: nn.Module, path: str):
     path = path.split(".")
     for name in path:
         module = getattr(module, name)
     return module
 
 
-def _set_module_by_path(module: "pl.LightningModule", path: str, value: Parameter):
+def _set_module_by_path(module: nn.Module, path: str, value: Parameter):
     path = path.split(".")
     for name in path[:-1]:
         module = getattr(module, name)