using walrus operator to simplify logic in GraphCreatorFactory

Frix-x · Nov 23, 2024 · f3c1917 · f3c1917
1 parent a9a0a3e
commit f3c1917
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/shaketune/graph_creators/graph_creator_factory.py b/shaketune/graph_creators/graph_creator_factory.py
@@ -14,7 +14,7 @@
 class GraphCreatorFactory:
     @staticmethod
     def create_graph_creator(graph_type: str, config: ShakeTuneConfig) -> GraphCreator:
-        creator_class = GraphCreator.registry.get(graph_type, None)
-        if not creator_class:
+        if creator_class := GraphCreator.registry.get(graph_type):
+            return creator_class(config)
+        else:
             raise NotImplementedError(f'Graph creator for {graph_type} not implemented!')
-        return creator_class(config)