From a3a77bd8797ed650d27be484c83a1fdd1ece577a Mon Sep 17 00:00:00 2001 From: LIU ZHE YOU Date: Wed, 5 Feb 2025 20:35:23 +0800 Subject: [PATCH] Merge fix get_classes_from_file #46454 --- tests/always/test_project_structure.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/tests/always/test_project_structure.py b/tests/always/test_project_structure.py index 0037ef4395a55..c4a45709f6313 100644 --- a/tests/always/test_project_structure.py +++ b/tests/always/test_project_structure.py @@ -248,8 +248,6 @@ def list_of_classes(self): for operator_file in self.new_class_paths(): operators_paths = self.get_classes_from_file(operator_file, NEW_PROVIDER_SRC, is_new=True) classes.update(operators_paths) - # replace src. prefix with empty string for all classes - classes = {k.strip("src."): v for k, v in classes.items()} return classes def get_classes_from_file( @@ -268,7 +266,8 @@ def get_classes_from_file( continue if is_new: - results[f"{'.'.join(module.split('.')[2:])}.{current_node.name}"] = current_node + module_path = module[module.find("airflow.providers") :] + results[f"{module_path}.{current_node.name}"] = current_node else: results[f"{module}.{current_node.name}"] = current_node print(f"{results}")