From 26738370c75ceaeafcc9c4c353f02a30d7dedd19 Mon Sep 17 00:00:00 2001 From: Yinon Polak Date: Tue, 4 Apr 2023 12:12:02 +0300 Subject: [PATCH] pytorch mlp add explicit annotation to fix mypy error --- freqtrade/freqai/torch/PyTorchMLPModel.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/freqtrade/freqai/torch/PyTorchMLPModel.py b/freqtrade/freqai/torch/PyTorchMLPModel.py index 192d2ad89..94cfa8e64 100644 --- a/freqtrade/freqai/torch/PyTorchMLPModel.py +++ b/freqtrade/freqai/torch/PyTorchMLPModel.py @@ -48,7 +48,7 @@ class PyTorchMLPModel(nn.Module): self.dropout = nn.Dropout(p=dropout_percent) def forward(self, x: List[torch.Tensor]) -> torch.Tensor: - x = x[0] + x: torch.Tensor = x[0] x = self.relu(self.input_layer(x)) x = self.dropout(x) x = self.blocks(x)