fixup! fixup! Adds support for Hook discovery from providers

apache · Nov 29, 2020 · 034c19e · 034c19e
1 parent a6cf883
commit 034c19e
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/airflow/plugins_manager.py b/airflow/plugins_manager.py
@@ -30,7 +30,7 @@
 import importlib_metadata
 
 from airflow import settings
-from airflow.utils.entry_points_with_dist import entry_points_with_dist
+from airflow.utils.entry_points import entry_points_with_dist
 from airflow.utils.file import find_path_from_directory
 
 if TYPE_CHECKING:

diff --git a/airflow/providers_manager.py b/airflow/providers_manager.py
@@ -27,7 +27,7 @@
 import jsonschema
 import yaml
 
-from airflow.utils.entry_points_with_dist import entry_points_with_dist
+from airflow.utils.entry_points import entry_points_with_dist
 
 try:
     import importlib.resources as importlib_resources
@@ -77,7 +77,7 @@ def __init__(self):
         self._discover_all_providers_from_packages()
         self._discover_hooks()
         self._provider_dict = OrderedDict(sorted(self.providers.items()))
-        self._hooks_dict = OrderedDict(sorted(self.providers.items()))
+        self._hooks_dict = OrderedDict(sorted(self.hooks.items()))
 
     def _discover_all_providers_from_packages(self) -> None:
         """

diff --git a/airflow/utils/entry_points_with_dist.py → airflow/utils/entry_points.py b/airflow/utils/entry_points_with_dist.py → airflow/utils/entry_points.py