mirror of
https://github.com/wassname/pytorch-ts.git
synced 2026-06-27 19:16:14 +08:00
Dr. Kashif Rasul
38 lines
887 B
Python
38 lines
887 B
Python
from setuptools import setup, find_packages
|
|
|
|
setup(
|
|
name='pytorchts',
|
|
version='0.2.0',
|
|
description="PyTorch Probabilistic Time Series Modeling framework",
|
|
long_description=open("README.md").read(),
|
|
long_description_content_type="text/markdown",
|
|
author='Kashif Rasul',
|
|
author_email="kashif.rasul@zalando.de",
|
|
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',
|
|
'glounts>=0.6.4',
|
|
'holidays',
|
|
'numpy',
|
|
'pandas>=1.1',
|
|
'scipy',
|
|
'tqdm',
|
|
'pydantic',
|
|
'matplotlib',
|
|
'tensorboard',
|
|
'wandb',
|
|
],
|
|
|
|
test_suite='tests',
|
|
tests_require = [
|
|
'flake8',
|
|
'pytest'
|
|
],
|
|
)
|