From 0c4574b3b7e2dac0ec2d441e2bc8bccf4c3ff9f5 Mon Sep 17 00:00:00 2001 From: Yinon Polak Date: Mon, 3 Apr 2023 18:10:47 +0300 Subject: [PATCH] prevent mypy error, explicitly unpack input list of pytorch mlp model, --- 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 01192e115..192d2ad89 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 + x = x[0] x = self.relu(self.input_layer(x)) x = self.dropout(x) x = self.blocks(x)