tidy up some type annotations (#5869)

python-poetry · Jun 18, 2022 · 84b7571 · 84b7571
1 parent 41a8a47
commit 84b7571
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 30 deletions.
diff --git a/src/poetry/puzzle/provider.py b/src/poetry/puzzle/provider.py
@@ -221,16 +221,7 @@ def search_for_installed_packages(
         )
         return packages
 
-    def search_for(
-        self,
-        dependency: (
-            Dependency
-            | VCSDependency
-            | FileDependency
-            | DirectoryDependency
-            | URLDependency
-        ),
-    ) -> list[DependencyPackage]:
+    def search_for(self, dependency: Dependency) -> list[DependencyPackage]:
         """
         Search for the specifications that match the given dependency.
 

diff --git a/src/poetry/puzzle/solver.py b/src/poetry/puzzle/solver.py
@@ -24,12 +24,8 @@
 
     from cleo.io.io import IO
     from poetry.core.packages.dependency import Dependency
-    from poetry.core.packages.directory_dependency import DirectoryDependency
-    from poetry.core.packages.file_dependency import FileDependency
     from poetry.core.packages.package import Package
     from poetry.core.packages.project_package import ProjectPackage
-    from poetry.core.packages.url_dependency import URLDependency
-    from poetry.core.packages.vcs_dependency import VCSDependency
 
     from poetry.puzzle.transaction import Transaction
     from poetry.repositories import Pool
@@ -248,22 +244,8 @@ def __init__(
         package: Package,
         packages: list[Package],
         previous: PackageNode | None = None,
-        previous_dep: None
-        | (
-            DirectoryDependency
-            | FileDependency
-            | URLDependency
-            | VCSDependency
-            | Dependency
-        ) = None,
-        dep: None
-        | (
-            DirectoryDependency
-            | FileDependency
-            | URLDependency
-            | VCSDependency
-            | Dependency
-        ) = None,
+        previous_dep: Dependency | None = None,
+        dep: Dependency | None = None,
     ) -> None:
         self.package = package
         self.packages = packages