bug fixes in NBEATSEstimator (#74)

shashankdeshpande · web-flow · commit 0b0dfe12cadd · 2021-09-02T09:53:09.000+02:00
* spell correction in n_beats_ensemble

* bug fix - NBEATSEstimator intialization
diff --git 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
@@ -30,7 +30,7 @@
 )
 
 
-class NBEATSEstimator:
+class NBEATSEstimator(PyTorchEstimator):
     @validated()
     def __init__(
         self,

Original file line number	Diff line number	Diff line change
`@@ -30,7 +30,7 @@`
`30`	`30`	`)`
`31`	`31`
`32`	`32`
`33`		`-class NBEATSEstimator:`
	`33`	`+class NBEATSEstimator(PyTorchEstimator):`
`34`	`34`	`@validated()`
`35`	`35`	`def __init__(`
`36`	`36`	`self,`