fix tensor type hint

2023-03-08 16:16:49 +02:00
parent c8296ccb2d
commit 7d26df01b8
1 changed files with 1 additions and 1 deletions
--- a/freqtrade/freqai/prediction_models/PyTorchClassifierMultiTarget.py
+++ b/freqtrade/freqai/prediction_models/PyTorchClassifierMultiTarget.py
@@ -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
        """