mirror of
https://github.com/wassname/pytorch-ts.git
synced 2026-06-27 19:16:14 +08:00
Kashif Rasul
34 lines
941 B
Python
34 lines
941 B
Python
from setuptools import setup, find_packages
|
|
|
|
setup(
|
|
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_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",
|
|
"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"],
|
|
)
|