CTYPTO/freqtrade
1
0
Fork 0
mirror of https://github.com/freqtrade/freqtrade.git synced 2026-02-05 22:00:25 +00:00
Code Issues 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

2
freqtrade/freqai/torch/PyTorchModelTrainer.py
View File

@@ -176,7 +176,7 @@ class PyTorchModelTrainer(PyTorchTrainerInterface):
"pytrainer": self
}, 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)
return self.load_from_checkpoint(checkpoint)