Fix minor error in fit return

LarsKue · Oct 27, 2023 · fa842a3 · fa842a3
1 parent eb29b16
commit fa842a3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/lightning_trainable/trainable/trainable.py b/src/lightning_trainable/trainable/trainable.py
@@ -269,7 +269,7 @@ def fit(self, logger_kwargs: dict = None, trainer_kwargs: dict = None, fit_kwarg
         return {
             key: value.item()
             for key, value in trainer.callback_metrics.items()
-            if any(key.startswith(key) for key in ["training/", "validation/"])
+            if any(key.startswith(k) for k in ["training/", "validation/"])
         }
 
     @torch.enable_grad()