formatting

pts/model/n_beats/n_beats_estimator.py

@@ -30,7 +30,7 @@ from .n_beats_network import (
 )
 
 
-class NBEATSEstimato
+class NBEATSEstimator:
     @validated()
     def __init__(
         self,

setup.py

@@ -1,39 +1,33 @@
 from setuptools import setup, find_packages
 
 setup(
-    name='pytorchts',
-    version='0.3.0',
+    name="pytorchts",
+    version="0.3.0",
     description="PyTorch Probabilistic Time Series Modeling framework",
     long_description=open("README.md").read(),
     long_description_content_type="text/markdown",
-    author='Kashif Rasul',
+    author="Kashif Rasul",
     author_email="kashif.rasul@zalando.de",
-    url='https://github.com/zalandoresearch/pytorch-ts',
-    license='MIT',
-
+    url="https://github.com/zalandoresearch/pytorch-ts",
+    license="MIT",
     packages=find_packages(exclude=["tests"]),
     include_package_data=True,
     zip_safe=True,
     python_requires=">=3.6",
-    install_requires = [
-        'torch>=1.7.0',
-        'gluonts',
-        'holidays',
-        'numpy',
-        'pandas>=1.1',
-        'scipy',
-        'tqdm',
-        'pydantic',
-        'matplotlib',
-        'tensorboard',
-        'wandb',
-    ],
-    dependency_links=[
-        'git+https://github.com/awslabs/gluon-ts.git@master#egg=gluonts'
-    ],
-    test_suite='tests',
-    tests_require = [
-        'flake8',
-        'pytest'
+    install_requires=[
+        "torch>=1.7.0",
+        "gluonts",
+        "holidays",
+        "numpy",
+        "pandas>=1.1",
+        "scipy",
+        "tqdm",
+        "pydantic",
+        "matplotlib",
+        "tensorboard",
+        "wandb",
     ],
+    dependency_links=["git+https://github.com/awslabs/gluon-ts.git@master#egg=gluonts"],
+    test_suite="tests",
+    tests_require=["flake8", "pytest"],
 )