suppress mypy error message when assigning Proxy to Parameter

Essoz · Essoz · commit 4bcee01bc2b4 · 2025-04-04T00:04:55.000-04:00
diff --git a/mldaikon/proxy_wrapper/proxy.py b/mldaikon/proxy_wrapper/proxy.py
@@ -114,7 +114,7 @@ def proxy_parameters(module: torch.nn.Module, parent_name="", from_iter=False):
         num_params = 0
         for name, parameter in module.named_parameters(recurse=False):
             num_params += 1
-            parameter = Proxy(
+            parameter = Proxy(  # type: ignore
                 parameter, var_name=parent_name + "." + name, from_iter=from_iter
             )
             module._parameters[name] = parameter

Original file line number	Diff line number	Diff line change
`@@ -114,7 +114,7 @@ def proxy_parameters(module: torch.nn.Module, parent_name="", from_iter=False):`
`114`	`114`	`num_params = 0`
`115`	`115`	`for name, parameter in module.named_parameters(recurse=False):`
`116`	`116`	`num_params += 1`
`117`		`- parameter = Proxy(`
	`117`	`+ parameter = Proxy( # type: ignore`
`118`	`118`	`parameter, var_name=parent_name + "." + name, from_iter=from_iter`
`119`	`119`	`)`
`120`	`120`	`module._parameters[name] = parameter`