mirror of
https://github.com/wassname/pytorch-ts.git
synced 2026-06-27 19:32:05 +08:00
Dr. Kashif Rasul
34 lines
761 B
Python
34 lines
761 B
Python
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.3.0',
|
|
'holidays',
|
|
'numpy',
|
|
'pandas',
|
|
'tqdm',
|
|
'ujson',
|
|
'pydantic',
|
|
'matplotlib',
|
|
],
|
|
|
|
test_suite='tests',
|
|
tests_require = [
|
|
'flake8',
|
|
'pytest'
|
|
],
|
|
)
|