fix comment predict

meteofrance · Jan 28, 2025 · c927651 · c927651
1 parent bba7578
commit c927651
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/py4cast/lightning.py b/py4cast/lightning.py
@@ -638,7 +638,7 @@ def on_load_checkpoint(self, checkpoint):
     def predict_step(self, batch: ItemBatch, batch_idx: int) -> torch.Tensor:
         """
         Check if the feature names are the same as the one used during training
-        and make a prediction and accumulate if the dataset used is poesy.
+        and make a prediction and accumulate if io_conf =/= none.
         """
         if batch_idx == 0:
             if self.input_feature_names != batch.inputs.feature_names: