From 0b0dfe12cadd2be3e0d4b9b146e4cfd2fdf42163 Mon Sep 17 00:00:00 2001
From: Shashank Deshpande <shashankdeshpande18@gmail.com>
Date: Thu, 2 Sep 2021 13:23:09 +0530
Subject: [PATCH] bug fixes in NBEATSEstimator (#74)

* spell correction in n_beats_ensemble

* bug fix - NBEATSEstimator intialization
---
 pts/model/n_beats/n_beats_ensemble.py  | 2 +-
 pts/model/n_beats/n_beats_estimator.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/pts/model/n_beats/n_beats_ensemble.py b/pts/model/n_beats/n_beats_ensemble.py
index f32cbbf..7792e90 100644
--- a/pts/model/n_beats/n_beats_ensemble.py
+++ b/pts/model/n_beats/n_beats_ensemble.py
@@ -170,7 +170,7 @@ class NBEATSEnsembleEstimator(PyTorchEstimator):
         Arguments passed down to the individual estimators.
     """
 
-    @validted()
+    @validated()
     def __init__(
         self,
         freq: str,
diff --git a/pts/model/n_beats/n_beats_estimator.py b/pts/model/n_beats/n_beats_estimator.py
index 40f3e5b..a00739e 100644
--- a/pts/model/n_beats/n_beats_estimator.py
+++ b/pts/model/n_beats/n_beats_estimator.py
@@ -30,7 +30,7 @@ from .n_beats_network import (
 )
 
 
-class NBEATSEstimator:
+class NBEATSEstimator(PyTorchEstimator):
     @validated()
     def __init__(
         self,