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
Show Stats 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}"
)
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
"""