Strip src. prefix instead of replacing all src.

Co-authored-by: Kalyan R <[email protected]>
jason810496 · Feb 5, 2025 · 0272798 · 0272798
1 parent bfd5d72
commit 0272798
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/always/test_project_structure.py b/tests/always/test_project_structure.py
@@ -249,7 +249,7 @@ def list_of_classes(self):
             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.replace("src.", ""): v for k, v in classes.items()}
+        classes = {k.strip("src."): v for k, v in classes.items()}
         return classes
 
     def get_classes_from_file(