mohammadian7/freqtrade
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

pytorch - mypy fixes

Browse Source
This commit is contained in:
Yinon Polak
2023-07-13 21:36:14 +03:00
parent 9fb0ce664c
commit ffcba45b1b
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
freqtrade/freqai/torch/PyTorchModelTrainer.py
+1 -1
View File
@@ -176,7 +176,7 @@ class PyTorchModelTrainer(PyTorchTrainerInterface):
"pytrainer": self "pytrainer": self
}, path) }, path)
def load(self, path: Path, device: str = None): def load(self, path: Path, device: Optional[str] = None):
checkpoint = torch.load(path, map_location=device) checkpoint = torch.load(path, map_location=device)
return self.load_from_checkpoint(checkpoint) return self.load_from_checkpoint(checkpoint)