resolves bug in lazy search where duplicated classes would occur (#218)

fidelity · Feb 1, 2022 · 3315430 · 3315430
1 parent 5e07103
commit 3315430
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/spock/graph.py b/spock/graph.py
@@ -110,7 +110,10 @@ def _lazily_find_classes(self, classes):
         # Iterate thorough all of the base spock classes to get the dependencies and reverse dependencies
         lazy_classes = []
         for _, v in self._yield_class_deps(classes):
-            if hasattr(sys.modules["spock"].backend.config, v):
+            if (
+                hasattr(sys.modules["spock"].backend.config, v)
+                and getattr(sys.modules["spock"].backend.config, v) not in classes
+            ):
                 print(
                     f"Lazy evaluation found a @spock decorated class named `{v}` within the registered types of "
                     f"sys.modules['spock'].backend.config -- Attempting to use the class "