adelorenzo/stable
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix tensor type hint

Browse Source
This commit is contained in:
Yinon Polak
2023-03-08 16:16:49 +02:00
parent c8296ccb2d
commit 7d26df01b8
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
freqtrade/freqai/prediction_models/PyTorchClassifierMultiTarget.py
View File

@@ -127,7 +127,7 @@ class PyTorchClassifierMultiTarget(BasePyTorchModel):
f"expecting labels: {self.multiclass_names}" f"expecting labels: {self.multiclass_names}"
) )
def decode_classes_name(self, classes: torch.Tensor[int]) -> List[str]: def decode_classes_name(self, classes: torch.Tensor) -> List[str]:
""" """
decode class name int -> str decode class name int -> str
""" """