Fixed test calls.

nomad-coe · May 3, 2024 · 2839be0 · 2839be0
1 parent 5ac17eb
commit 2839be0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/simulationworkflownormalizer/normalizer.py b/simulationworkflownormalizer/normalizer.py
@@ -75,7 +75,7 @@ def _resolve_workflow(self, archive: EntryArchive):
 
     def normalize(self, archive: EntryArchive, logger=None) -> None:
         logger = logger if logger is not None else get_logger(__name__)
-        super().normalize(logger)
+        super().normalize(archive, logger)
 
         # Do nothing if run section is not present
         if not archive.run:

diff --git a/tests/test_simulationworkflownormalizer.py b/tests/test_simulationworkflownormalizer.py
@@ -46,7 +46,7 @@ def test_resolve_workflow_from_program_name(
 ):
     run = Run(program=Program(name=program_name))
     entry_archive.run.append(run)
-    SimulationWorkflowNormalizer(entry_archive).normalize(get_logger(__name__))
+    SimulationWorkflowNormalizer().normalize(entry_archive, get_logger(__name__))
     assert isinstance(entry_archive.workflow2, workflow_class)
 
 
@@ -58,5 +58,5 @@ def test_resolve_workflow_from_calculation(
 ):
     run = Run(calculation=[Calculation() for _ in range(n_calculations)])
     entry_archive.run.append(run)
-    SimulationWorkflowNormalizer(entry_archive).normalize(get_logger(__name__))
+    SimulationWorkflowNormalizer().normalize(entry_archive, get_logger(__name__))
     assert isinstance(entry_archive.workflow2, workflow_class)