setup.py

# -*- coding: utf-8 -*-

# License: BSD 3-Clause

import os
import setuptools
import sys

if sys.version_info < (3, 5):
    raise ValueError(
        'Unsupported Python version {}.{}.{} found. OpenML requires Python 3.5 or higher.'
        .format(sys.version_info.major, sys.version_info.minor, sys.version_info.micro)
    )

with open(os.path.join("README.md")) as fid:
    README = fid.read()

setuptools.setup(name="openml-pytorch",
                 author=" ",
                 version ="0.0.6",
                 author_email="",
                 maintainer="Taniya Das",
                 maintainer_email="t.das@tue.nl",
                 description="Pytorch extension for Openml python API",
                 long_description=README,
                 long_description_content_type='text/markdown',
                 license="BSD 3-clause",
                 url="http://openml.org/",
                 project_urls={
                     "Documentation": "https://openml.github.io/openml-pytorch/",
                     "Source Code": "https://github.com/openml/openml-pytorch"
                 },
                 # Make sure to remove stale files such as the egg-info before updating this:
                 # https://stackoverflow.com/a/26547314
                 packages=setuptools.find_packages(
                     include=['openml_pytorch.*', 'openml_pytorch'],
                     exclude=["*.tests", "*.tests.*", "tests.*", "tests"],
                 ),
                #  python 3.12 have removed distutils, so for now, we only support till python <3.11
                 package_data={'': ['*.txt', '*.md']},
                 python_requires=">=3.5, <3.11",
                 install_requires=[
                     'openml',
                     'torch>=1.4.0, <2.2.0',
                     'onnx',
                     'torchvision',
                 ],
                 classifiers=['Intended Audience :: Science/Research',
                              'Intended Audience :: Developers',
                              'License :: OSI Approved :: BSD License',
                              'Programming Language :: Python',
                              'Topic :: Software Development',
                              'Topic :: Scientific/Engineering',
                              'Operating System :: POSIX',
                              'Operating System :: Unix',
                              'Operating System :: MacOS',
                              'Programming Language :: Python :: 3',
                              'Programming Language :: Python :: 3.5',
                              'Programming Language :: Python :: 3.6',
                              'Programming Language :: Python :: 3.7'])