File size: 1,647 Bytes
2d1810a | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 | from setuptools import setup, find_packages
import io
import os
here = os.path.abspath(os.path.dirname(__file__))
long_description = """
gconvex: Finitely convex parametrization of generalized convex functions
and experimental optimal_transport for Monge map / optimal transport comparisons.
"""
try:
with io.open(os.path.join(here, "README.md"), encoding="utf-8") as f:
long_description = f.read()
except Exception:
pass
setup(
name="gconvex",
version="0.1.0",
description="Finitely convex parametrization and OT optimal_transport",
long_description=long_description,
long_description_content_type="text/markdown",
author="Moeen Nehzati",
url="https://github.com/MoeenNehzati/gconvex",
packages=find_packages(exclude=("tests", "docs")),
include_package_data=True,
python_requires=">=3.10",
install_requires=[
# Core numeric / ML (torch>=2.0 needed for torch.compile)
"torch>=2.0.0,<3.0",
"numpy>=1.19.0,<2.0",
# Visualization & data
"matplotlib>=3.3.0,<4.0",
"pandas>=1.1.0,<2.0",
# Logging / notebook utilities
"rich>=10.0.0",
"ipython>=7.0.0",
],
extras_require={
"dev": [
"pytest>=7.0.0",
"pytest-xdist>=3.0.0", # Parallel test execution
"pytest-timeout>=2.0.0", # Test timeouts
"flake8",
],
"wandb": ["wandb"],
},
license="Apache-2.0",
classifiers=[
"Programming Language :: Python :: 3",
"License :: OSI Approved :: Apache Software License",
"Operating System :: OS Independent",
],
)
|