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

Don't use assert in non-test code.

Browse Source
This commit is contained in:
Matthias
2024-06-08 09:32:54 +02:00
parent cef9c45f68
commit de5a5d0967
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
freqtrade/freqai/torch/PyTorchModelTrainer.py
+2 -1
View File
@@ -148,7 +148,8 @@ class PyTorchModelTrainer(PyTorchTrainerInterface):
the motivation here is that `n_steps` is easier to optimize and keep stable, the motivation here is that `n_steps` is easier to optimize and keep stable,
across different n_obs - the number of data points. across different n_obs - the number of data points.
""" """
assert isinstance(self.n_steps, int), "Either `n_steps` or `n_epochs` should be set." if not isinstance(self.n_steps, int):
raise ValueError("Either `n_steps` or `n_epochs` should be set.")
n_batches = n_obs // self.batch_size n_batches = n_obs // self.batch_size
n_epochs = max(self.n_steps // n_batches, 1) n_epochs = max(self.n_steps // n_batches, 1)
if n_epochs <= 10: if n_epochs <= 10: