From a6555242218ddbf7269353baad69e0472531b61d Mon Sep 17 00:00:00 2001 From: Yinon Polak Date: Tue, 4 Apr 2023 12:24:29 +0300 Subject: [PATCH] pytorch mlp rename input to fix mypy error --- freqtrade/freqai/torch/PyTorchMLPModel.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/freqtrade/freqai/torch/PyTorchMLPModel.py b/freqtrade/freqai/torch/PyTorchMLPModel.py index 94cfa8e64..e50ccd5ef 100644 --- a/freqtrade/freqai/torch/PyTorchMLPModel.py +++ b/freqtrade/freqai/torch/PyTorchMLPModel.py @@ -47,8 +47,8 @@ class PyTorchMLPModel(nn.Module): self.relu = nn.ReLU() self.dropout = nn.Dropout(p=dropout_percent) - def forward(self, x: List[torch.Tensor]) -> torch.Tensor: - x: torch.Tensor = x[0] + def forward(self, tensors: List[torch.Tensor]) -> torch.Tensor: + x: torch.Tensor = tensors[0] x = self.relu(self.input_layer(x)) x = self.dropout(x) x = self.blocks(x)