From b053285fe91d8566dc002a0945e91fe9769a3220 Mon Sep 17 00:00:00 2001
From: "Dr. Kashif Rasul" <kashif.rasul@gmail.com>
Date: Mon, 10 Feb 2020 13:22:41 +0100
Subject: [PATCH] no need for input_size its always 1

---
 pts/model/simple_feedforward/simple_feedforward_estimator.py | 4 ----
 pts/model/simple_feedforward/simple_feedforward_network.py   | 2 --
 2 files changed, 6 deletions(-)

diff --git a/pts/model/simple_feedforward/simple_feedforward_estimator.py b/pts/model/simple_feedforward/simple_feedforward_estimator.py
index 06202b4..50adda2 100644
--- a/pts/model/simple_feedforward/simple_feedforward_estimator.py
+++ b/pts/model/simple_feedforward/simple_feedforward_estimator.py
@@ -77,7 +77,6 @@ class SimpleFeedForwardEstimator(PTSEstimator):
         self,
         freq: str,
         prediction_length: int,
-        input_size: int,
         trainer: Trainer = Trainer(),
         num_hidden_dimensions: Optional[List[int]] = None,
         context_length: Optional[int] = None,
@@ -96,7 +95,6 @@ class SimpleFeedForwardEstimator(PTSEstimator):
             if num_hidden_dimensions is not None
             else list([40, 40])
         )
-        self.input_size = input_size
         self.prediction_length = prediction_length
         self.context_length = (
             context_length if context_length is not None else prediction_length
@@ -136,7 +134,6 @@ class SimpleFeedForwardEstimator(PTSEstimator):
         self, device: torch.device
     ) -> SimpleFeedForwardTrainingNetwork:
         return SimpleFeedForwardTrainingNetwork(
-            input_size=self.input_size,
             num_hidden_dimensions=self.num_hidden_dimensions,
             prediction_length=self.prediction_length,
             context_length=self.context_length,
@@ -154,7 +151,6 @@ class SimpleFeedForwardEstimator(PTSEstimator):
         device: torch.device,
     ) -> PTSPredictor:
         prediction_network = SimpleFeedForwardPredictionNetwork(
-            input_size=self.input_size,
             num_hidden_dimensions=self.num_hidden_dimensions,
             prediction_length=self.prediction_length,
             context_length=self.context_length,
diff --git a/pts/model/simple_feedforward/simple_feedforward_network.py b/pts/model/simple_feedforward/simple_feedforward_network.py
index c76b06d..ca56059 100644
--- a/pts/model/simple_feedforward/simple_feedforward_network.py
+++ b/pts/model/simple_feedforward/simple_feedforward_network.py
@@ -38,7 +38,6 @@ class SimpleFeedForwardNetworkBase(nn.Module):
 
     def __init__(
         self,
-        input_size: int,
         num_hidden_dimensions: List[int],
         prediction_length: int,
         context_length: int,
@@ -48,7 +47,6 @@ class SimpleFeedForwardNetworkBase(nn.Module):
     ) -> None:
         super().__init__()
 
-        self.input_size = input_size
         self.num_hidden_dimensions = num_hidden_dimensions
         self.prediction_length = prediction_length
         self.context_length = context_length