Fix code for mypy

python-poetry · Aug 23, 2023 · 35be3e7 · 35be3e7
1 parent b243837
commit 35be3e7
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/src/poetry/core/factory.py b/src/poetry/core/factory.py
@@ -11,6 +11,7 @@
 from typing import Union
 from warnings import warn
 
+from packaging.utils import NormalizedName
 from packaging.utils import canonicalize_name
 
 from poetry.core.utils.helpers import combine_unicode
@@ -179,20 +180,24 @@ def configure_package(
                 package=package, group="dev", dependencies=config["dev-dependencies"]
             )
 
-        extra_deps = {}
+        extra_deps: dict[
+            tuple[NormalizedName, frozenset[NormalizedName]], Dependency
+        ] = {}
         extras = config.get("extras", {})
         for extra_name, requirements in extras.items():
             extra_name = canonicalize_name(extra_name)
             package.extras[extra_name] = []
 
             # Checking for dependency
             for req in requirements:
+                req_extras: list[NormalizedName] = []
                 req_match = REQUIREMENT_PATTERN.match(req.replace(" ", ""))
                 if req_match is not None:
                     req = req_match.group(1)
-                    req_extras = req_match.group(2).split(",")
-                else:
-                    req_extras = []
+                    req_extras.extend(
+                        canonicalize_name(extra)
+                        for extra in req_match.group(2).split(",")
+                    )
 
                 req = Dependency(req, "*")