diff --git a/CI/unit_tests/recorders/test_base_recorder.py b/CI/unit_tests/recorders/test_base_recorder.py index 368243d..e0160ef 100644 --- a/CI/unit_tests/recorders/test_base_recorder.py +++ b/CI/unit_tests/recorders/test_base_recorder.py @@ -190,7 +190,7 @@ def test_store(self): # Test storing recorder._measure(**self.neural_state) - recorder._store() + recorder.store() data = recorder.load() assert set(data.keys()) == {"dummy_1", "dummy_2"} @@ -199,7 +199,7 @@ def test_store(self): # Test storing again recorder._measure(**self.neural_state) - recorder._store() + recorder.store() data = recorder.load() assert set(data.keys()) == {"dummy_1", "dummy_2"} @@ -211,7 +211,7 @@ def test_store(self): # test overwriting recorder.overwrite = True recorder._measure(**self.neural_state) - recorder._store() + recorder.store() data = recorder.load() assert set(data.keys()) == {"dummy_1", "dummy_2"} @@ -243,11 +243,11 @@ def test_counter(self): assert recorder._counter == 1 recorder._measure(**self.neural_state) assert recorder._counter == 2 - recorder._store() # It should not story due to the chunk size + recorder.store() # It should not story due to the chunk size assert recorder._counter == 2 recorder._measure(**self.neural_state) assert recorder._counter == 3 - recorder._store() # It should store now + recorder.store() # It should store now assert recorder._counter == 0 # Delete temporary directory diff --git a/papyrus/recorders/base_recorder.py b/papyrus/recorders/base_recorder.py index 3f93123..b650ba1 100644 --- a/papyrus/recorders/base_recorder.py +++ b/papyrus/recorders/base_recorder.py @@ -173,7 +173,7 @@ def _measure(self, **neural_state): # Increment the counter self._counter += 1 - def _store(self): + def store(self): """ Store the results of the measurements in the database. @@ -221,4 +221,4 @@ def record(self, neural_state: dict): The result of the recorder. """ self._measure(**neural_state) - self._store() + self.store()