fix learning test

molmod · Nov 1, 2023 · bd43827 · bd43827
1 parent 1425141
commit bd43827
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/psiflow/learning.py b/psiflow/learning.py
@@ -34,7 +34,7 @@ class BaseLearning:
     pretraining_nstates: int = 50
     pretraining_amplitude_pos: float = 0.05
     pretraining_amplitude_box: float = 0.0
-    metrics: Metrics = Metrics()
+    metrics: Metrics = field(default_factory=lambda: Metrics())  # mutable default
     atomic_energies: dict[str, Union[float, AppFuture]] = field(
         default_factory=lambda: {}
     )

diff --git a/tests/test_learning.py b/tests/test_learning.py
@@ -16,7 +16,6 @@ def test_learning_save_load(gpu, tmp_path):
     path_output.mkdir()
     SequentialLearning(
         path_output=path_output,
-        metrics=None,
         pretraining_nstates=100,
     )
     learning_ = load_learning(path_output)