From 945f830207904b1ba06b277d42975fdcc82b3a85 Mon Sep 17 00:00:00 2001
From: Kashif Rasul <kashif.rasul@zalando.de>
Date: Sun, 14 Jul 2019 14:34:41 +0200
Subject: [PATCH] initial setup.py

---
 setup.py | 32 ++++++++++++++++++++++++++++++++
 1 file changed, 32 insertions(+)
 create mode 100644 setup.py

diff --git a/setup.py b/setup.py
new file mode 100644
index 0000000..0c2abd5
--- /dev/null
+++ b/setup.py
@@ -0,0 +1,32 @@
+from setuptools import setup, find_packages
+
+setup(
+    name='pytorch-ts',
+    version='0.1.0',
+    description="PyTorch Probabilistic Time Series Modeling framework",
+    long_description=open("README.md").read(),
+    long_description_content_type="text/markdown",
+
+    url='https://github.com/kashif/pytorch-ts',
+    license='Apache License Version 2.0',
+
+    packages=find_packages(exclude=["tests"]),
+    include_package_data=True,
+    zip_safe=True,
+    python_requires=">=3.6",
+    install_requires = [
+        'torch>=1.0.0',
+        'holidays',
+        'numpy',
+        'pandas',
+        'tqdm',
+        'ujson',
+        'pydantic',
+    ],
+
+    test_suite='tests',
+    tests_require = [
+        'flake8',
+        'pytest'
+    ],
+)