Make sure graph is not generated by AttackGraph.__init__ when deepcop…

…ying
mal-lang · Sep 17, 2024 · 05cd8c8 · 05cd8c8
1 parent 5bdf57e
commit 05cd8c8
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/maltoolbox/attackgraph/attackgraph.py b/maltoolbox/attackgraph/attackgraph.py
@@ -226,7 +226,8 @@ def _to_dict(self) -> dict:
         }
 
     def __deepcopy__(self, memo):
-        copied_attackgraph = AttackGraph(self.lang_graph, self.model)
+        copied_attackgraph = AttackGraph(self.lang_graph)
+        copied_attackgraph.model = self.model
 
         # Deep copy nodes and add references to them
         copied_attackgraph.nodes = copy.deepcopy(self.nodes, memo)