setup.py

#!/usr/bin/env python
#  Copyright (c) Microsoft Corporation.
#  Licensed under the MIT license.
from setuptools import find_packages, setup

setup(
    name="sm-dataflow",
    version="0.1",
    author="Semantic Machines (TM)",
    description="Task-Oriented Dialogue as Dataflow Synthesis",
    license="MIT",
    packages=find_packages("src"),
    package_dir={"": "src"},
    package_data={"dataflow": ["py.typed"]},
    zip_safe=False,
    install_requires=[
        "jsons==0.10.1",
        "pydantic==1.4",
        "more-itertools==8.2.0",
        "sexpdata==0.0.3",
        "pandas==1.0.0",
        "spacy==2.2.1",
        "statsmodels==0.11.1",
    ],
    extra_requires={
        "OpenNMT-py": ["OpenNMT-py==1.0.0", "pytorch>=1.2.0,<=1.4.0"]
    },
    python_requires=">=3.6",
)