update test

skorch-dev · Feb 24, 2023 · 909f4d5 · 909f4d5
1 parent 2e4952c
commit 909f4d5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/skorch/tests/callbacks/test_logging.py b/skorch/tests/callbacks/test_logging.py
@@ -163,7 +163,7 @@ def test_log_on_batch_level_on(
 
         # (5 epochs x (40/4 batches x 2 batch metrics + 2 epoch metrics) = 110 calls) + base metrics
         assert mock_experiment.__getitem__.call_count == 110 + self.NUM_BASE_METRICS
-        mock_experiment['training']['train']['batch']['batch_size'].log.assert_any_call(4)
+        mock_experiment['training']['train']['batch']['batch_size'].append.assert_any_call(4)
 
     def test_log_on_batch_level_off(
             self,