Correctly recognize typing_extensions.NewType

python · Oct 19, 2023 · 4882ccc · 4882ccc
1 parent 1f20d68
commit 4882ccc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mypy/semanal_newtype.py b/mypy/semanal_newtype.py
@@ -147,7 +147,7 @@ def analyze_newtype_declaration(self, s: AssignmentStmt) -> tuple[str | None, Ca
             and isinstance(s.lvalues[0], NameExpr)
             and isinstance(s.rvalue, CallExpr)
             and isinstance(s.rvalue.callee, RefExpr)
-            and s.rvalue.callee.fullname == "typing.NewType"
+            and (s.rvalue.callee.fullname in ("typing.NewType", "typing_extensions.NewType"))
         ):
             name = s.lvalues[0].name