| #!/usr/bin/env python |
| # -*- coding: utf-8 -*- |
| import warnings |
| |
| """The setup script.""" |
| |
| from setuptools import find_packages, setup |
| |
| # don't fail if there are problems with the readme (happens within circleci) |
| try: |
| with open("README.md") as readme_file: |
| readme = readme_file.read() |
| except Exception: |
| warnings.warn("README.md not found") |
| readme = None |
| |
| REQUIREMENTS_FILES = ["requirements.txt"] |
| |
| |
| def get_version(): |
| version_dict = {} |
| with open("hamilton/version.py") as f: |
| exec(f.read(), version_dict) |
| return ".".join(map(str, version_dict["VERSION"])) |
| |
| |
| VERSION = get_version() |
| |
| |
| def load_requirements(): |
| requirements = set() |
| for requirement_file in REQUIREMENTS_FILES: |
| with open(requirement_file) as f: |
| requirements.update(line.strip() for line in f) |
| return list(requirements) |
| |
| |
| setup( |
| name="sf-hamilton", # there's already a hamilton in pypi |
| version=VERSION, |
| description="Hamilton, the micro-framework for creating dataframes.", |
| long_description=readme, |
| long_description_content_type="text/markdown", |
| author="Stefan Krawczyk, Elijah ben Izzy", |
| author_email="stefan@dagworks.io,elijah@dagworks.io", |
| url="https://github.com/dagworks-inc/hamilton", |
| packages=find_packages(exclude=["tests"]), |
| include_package_data=True, |
| install_requires=load_requirements(), |
| zip_safe=False, |
| keywords="hamilton", |
| classifiers=[ |
| "Development Status :: 5 - Production/Stable", |
| "Intended Audience :: Developers", |
| "Natural Language :: English", |
| "License :: OSI Approved :: BSD License", |
| "Programming Language :: Python :: 3", |
| "Programming Language :: Python :: 3.7", |
| "Programming Language :: Python :: 3.8", |
| "Programming Language :: Python :: 3.9", |
| "Programming Language :: Python :: 3.10", |
| "Programming Language :: Python :: 3.11", |
| ], |
| # Note that this feature requires pep8 >= v9 and a version of setup tools greater than the |
| # default version installed with virtualenv. Make sure to update your tools! |
| python_requires=">=3.6, <4", |
| # adding this to slim the package down, since these dependencies are only used in certain contexts. |
| extras_require={ |
| "visualization": ["graphviz", "networkx"], |
| "dask": ["dask[complete]"], # commonly you'll want everything. |
| "dask-array": ["dask[array]"], |
| "dask-core": ["dask-core"], |
| "dask-dataframe": ["dask[dataframe]"], |
| "dask-diagnostics": ["dask[diagnostics]"], |
| "dask-distributed": ["dask[distributed]"], |
| "ray": ["ray>=2.0.0", "pyarrow"], |
| "pyspark": [ |
| # we have to run these dependencies cause Spark does not check to ensure the right target was called |
| "pyspark[pandas_on_spark,sql]", |
| # This is problematic, see https://stackoverflow.com/questions/76072664/convert-pyspark-dataframe-to-pandas-dataframe-fails-on-timestamp-column |
| "pandas<2.0", |
| ], # I'm sure they'll add support soon, |
| # but for now its not compatible |
| "pandera": ["pandera"], |
| }, |
| # Relevant project URLs |
| project_urls={ # Optional |
| "Bug Reports": "https://github.com/dagworks-inc/hamilton/issues", |
| "Source": "https://github.com/dagworks-inc/hamilton", |
| }, |
| ) |
