fix argumment name

2026-07-17 11:32:26 +08:00 · 2020-01-28 11:27:58 +01:00
parent 1ab2babbe3
commit 9322ce004a
2 changed files with 2 additions and 2 deletions
@@ -418,7 +418,7 @@ class TransformerPredictionNetwork(TransformerNetwork):
        self,
        feat_static_cat: torch.Tensor,
        feat_static_real: torch.Tensor,
-        feature_static_real: torch.Tensor,
+        feat_static_real: torch.Tensor,
        past_time_feat: torch.Tensor,
        past_target: torch.Tensor,
        past_observed_values: torch.Tensor,
@@ -441,7 +441,7 @@ class TransformerPredictionNetwork(TransformerNetwork):
        # create the inputs for the encoder
        inputs, scale, static_feat = self.create_network_input(
            feat_static_cat=feat_static_cat,
-            feature_static_real=feat_static_real,
+            feat_static_cat=feat_static_real,
            past_time_feat=past_time_feat,
            past_target=past_target,
            past_observed_values=past_observed_values,