Linter fixes

mal-lang · Mar 21, 2024 · be42b6f · be42b6f
1 parent 75aeb20
commit be42b6f
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/malpzsim/sims/mal_petting_zoo_simulator.py b/malpzsim/sims/mal_petting_zoo_simulator.py
@@ -586,9 +586,9 @@ def render(self):
         logger.debug("Ingest attack graph into Neo4J database.")
         neo4j.ingest_attack_graph(
             self.attack_graph,
-            maltoolbox.neo4j_configs["uri"],
-            maltoolbox.neo4j_configs["username"],
-            maltoolbox.neo4j_configs["password"],
-            maltoolbox.neo4j_configs["dbname"],
+            neo4j_configs["uri"],
+            neo4j_configs["username"],
+            neo4j_configs["password"],
+            neo4j_configs["dbname"],
             delete=True,
         )
diff --git a/tests/run_demo.py b/tests/run_demo.py
@@ -43,7 +43,7 @@ def default(self, o):
 lang_classes_factory = LanguageClassesFactory(lang_spec)
 lang_classes_factory.create_classes()
 
-lang_graph = mallanguagegraph.LanguageGraph(lang_spec)
+lang_graph = LanguageGraph(lang_spec)
 
 model = Model("Test Model", lang_spec, lang_classes_factory)
 model.load_from_file("tests/example_model.json")