This seems cleaner

aiidateam · Oct 10, 2023 · 9a8e308 · 9a8e308
1 parent d77e464
commit 9a8e308
Showing 1 changed file with 1 addition and 6 deletions.
diff --git a/aiida/plugins/entry_point.py b/aiida/plugins/entry_point.py
@@ -349,13 +349,8 @@ def get_entry_point_from_class(class_module: str, class_name: str) -> Tuple[Opti
     :return: a tuple of the corresponding group and entry point or None if not found
     """
     for entry_point in eps():
-
-        if entry_point.module != class_module:
-            continue
-
-        if entry_point.attr == class_name:
+        if entry_point.module == class_module and entry_point.attr == class_name:
             return entry_point.group, entry_point
-
     return None, None