Merge pull request #702 from NeilGirdhar:fix_pyright

PiperOrigin-RevId: 553138908
google-deepmind · Aug 2, 2023 · b7b150f · b7b150f
2 parents 3c8e044 + 0a36df9
commit b7b150f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/haiku/_src/module.py b/haiku/_src/module.py
@@ -124,7 +124,7 @@ def __new__(  # pylint: disable=bad-classmethod-argument
 
     return cls
 
-  def __call__(cls: Type[T], *args, **kwargs) -> T:  # pylint: disable=no-self-argument
+  def __call__(cls: "ModuleMetaclass", *args, **kwargs) -> Any:  # pylint: disable=no-self-argument
     # Call new such that we have an un-initialized module instance that we can
     # still reference even if there is an exception during __init__. This is
     # needed such that we can make sure the name_scope constructed in __init__