From 830a5df2aa741a3c3b521991ec9b1cbc992ecae3 Mon Sep 17 00:00:00 2001
From: "Dr. Kashif Rasul" <kashif.rasul@gmail.com>
Date: Wed, 11 Mar 2020 13:33:51 +0100
Subject: [PATCH] removed unused fields for now

---
 pts/model/n_beats/n_beats_estimator.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/pts/model/n_beats/n_beats_estimator.py b/pts/model/n_beats/n_beats_estimator.py
index 6c067fc..5dcd121 100644
--- a/pts/model/n_beats/n_beats_estimator.py
+++ b/pts/model/n_beats/n_beats_estimator.py
@@ -11,6 +11,7 @@ from pts.transform import (
     InstanceSplitter,
     Transformation,
     Chain,
+    RemoveFields,
     ExpectedNumInstanceSampler,
 )
 from pts.dataset import FieldName
@@ -126,7 +127,10 @@ class NBEATSEstimator(PTSEstimator):
     # conditioning part and a to-predict part, for each training example.
     def create_transformation(self) -> Transformation:
         return Chain(
-            [
+            [   RemoveFields(
+                    field_names=[FieldName.FEAT_STATIC_REAL, 
+                        FieldName.FEAT_DYNAMIC_REAL, 
+                        FieldName.FEAT_DYNAMIC_CAT]),
                 InstanceSplitter(
                     target_field=FieldName.TARGET,
                     is_pad_field=FieldName.IS_PAD,