type hints fixes

2023-03-06 19:14:54 +02:00
parent 125085fbaf
commit 8acdd0b47c
2 changed files with 3 additions and 2 deletions
--- a/freqtrade/freqai/base_models/PyTorchModelTrainer.py
+++ b/freqtrade/freqai/base_models/PyTorchModelTrainer.py
@@ -51,7 +51,7 @@ class PyTorchModelTrainer:
            # training
            for batch_data in data_loaders_dictionary['train']:
                xb, yb = batch_data
-                xb = xb.to(self.device)  # type: ignore
+                xb = xb.to(self.device)
                yb = yb.to(self.device)
                yb_pred = self.model(xb)
                loss = self.criterion(yb_pred, yb)