Adding missing logger line

Author: franchuterivera

test/test_pipeline/components/training/test_training.py

@@ -171,7 +171,7 @@ def test_regression_epoch_training(self):
         counter = 0
         r2 = 0
         while r2 < 0.7:
-            loss, metrics = trainer.train_epoch(loader, epoch=1, logger=logger, writer=None)
+            loss, metrics = trainer.train_epoch(loader, epoch=1, writer=None)
             counter += 1
             r2 = metrics['r2']