Skip to content

Commit

Permalink
Merge pull request #34 from Techtonique/v0200
Browse files Browse the repository at this point in the history 
do v0.20.1
  • Loading branch information
@thierrymoudiki
thierrymoudiki authored Oct 3, 2024
2 parents d558a03 + 177b736 commit 7fd285f
Showing 1 changed file with 174 additions and 73 deletions.
247 changes: 174 additions & 73 deletions setup.py
View file
Original file line number Diff line number Diff line change
@@ -1,106 +1,207 @@
#!/usr/bin/env python
#
# Copyright (C) 2020-2024 T. Moudiki <[email protected]>
# License: 3-clause BSD

import subprocess

subprocess.check_call(['pip', 'install', 'Cython==3.0.10'])

import os
import platform
import setuptools
import shutil
import sys
import subprocess

from os import path
from pathlib import Path
from setuptools import Extension, find_packages, setup

subprocess.check_call([sys.executable, "-m", "pip", "install", "numpy"])
subprocess.check_call([sys.executable, "-m", "pip", "install", "cython"])
try:
import builtins
except ImportError:
import __builtin__ as builtins

subprocess.run(['pip', 'install', 'numpy'], check=False)
subprocess.run(['pip', 'install', 'scipy'], check=False)
subprocess.run(['pip', 'install', 'requests'], check=False)

from Cython.Build import cythonize
import numpy
from Cython.Build import cythonize

builtins.__MLSAUCE_SETUP__ = True

DISTNAME = 'mlsauce'
DESCRIPTION = 'Miscellaneous Statistical/Machine Learning tools'
LONG_DESCRIPTION = 'Miscellaneous Statistical/Machine Learning tools'
MAINTAINER = 'T. Moudiki'
MAINTAINER_EMAIL = '[email protected]'
LICENSE = 'BSD3 Clause Clear'
VERSION = '0.20.0'

# Minimum versions for dependencies
__version__ = '0.20.1'

VERSION = __version__

if platform.python_implementation() == 'PyPy':
SCIPY_MIN_VERSION = '1.1.0'
NUMPY_MIN_VERSION = '1.14.0'
else:
SCIPY_MIN_VERSION = '0.19.0'
NUMPY_MIN_VERSION = '1.13.0'

install_requires = [
f"numpy >= {NUMPY_MIN_VERSION}",
f"scipy >= {SCIPY_MIN_VERSION}",
"joblib >= 1.2.0",
"scikit-learn >= 0.18.0",
"threadpoolctl >= 2.0.0",
"pandas >= 0.25.3",
"requests >= 2.31.0"
]

# Include JAX only on Linux and macOS
JOBLIB_MIN_VERSION = '1.2.0'
SKLEARN_MIN_VERSION = '0.18.0'
THREADPOOLCTL_MIN_VERSION = '2.0.0'
PANDAS_MIN_VERSION = '0.25.3'
QUERIER_MIN_VERSION = '0.4.0'
REQUESTS_MIN_VERSION = '2.31.0'
if platform.system() in ('Linux', 'Darwin'):
install_requires += ["jax >= 0.1.72", "jaxlib >= 0.1.51"]

# Define extensions
ext_modules = [
Extension(
"mlsauce.adaopt._adaoptc",
sources=["mlsauce/adaopt/_adaoptc.pyx"],
include_dirs=[numpy.get_include()]
),
Extension(
"mlsauce.booster._boosterc",
sources=["mlsauce/booster/_boosterc.pyx"],
include_dirs=[numpy.get_include()]
),
Extension(
"mlsauce.lasso._lassoc",
sources=["mlsauce/lasso/_lassoc.pyx"],
include_dirs=[numpy.get_include()]
),
Extension(
"mlsauce.ridge._ridgec",
sources=["mlsauce/ridge/_ridgec.pyx"],
include_dirs=[numpy.get_include()]
),
Extension(
"mlsauce.stump._stumpc",
sources=["mlsauce/stump/_stumpc.pyx"],
include_dirs=[numpy.get_include()]
),
]
JAX_MIN_VERSION = '0.1.72'
JAXLIB_MIN_VERSION = '0.1.51'

def setup_package():
metadata = dict(
name=DISTNAME,
maintainer=MAINTAINER,
maintainer_email=MAINTAINER_EMAIL,
description=DESCRIPTION,
license=LICENSE,
version=VERSION,
long_description=LONG_DESCRIPTION,
classifiers=[
'Development Status :: 2 - Pre-Alpha',
'Intended Audience :: Developers',
'License :: OSI Approved :: BSD License',
'Programming Language :: Python :: 3',
'Programming Language :: Python :: 3.5',
'Programming Language :: Python :: 3.6',
'Programming Language :: Python :: 3.7',
'Programming Language :: Python :: 3.8',
],
platforms=["linux", "macosx", "windows"],
python_requires=">=3.5",
install_requires=install_requires,
setup_requires=["numpy>=1.13.0"],
packages=find_packages(),
ext_modules=cythonize(ext_modules),
SETUPTOOLS_COMMANDS = {
'develop', 'release', 'bdist_egg', 'bdist_rpm',
'bdist_wininst', 'install_egg_info', 'build_sphinx',
'egg_info', 'easy_install', 'upload', 'bdist_wheel',
'--single-version-externally-managed',
}

if SETUPTOOLS_COMMANDS.intersection(sys.argv):
extra_setuptools_args = dict(
zip_safe=False, # the package can run out of an .egg file
include_package_data=True,
extras_require={
'alldeps': (
'numpy >= {}'.format(NUMPY_MIN_VERSION),
'scipy >= {}'.format(SCIPY_MIN_VERSION),
),
},
)
else:
extra_setuptools_args = dict()


WHEELHOUSE_UPLOADER_COMMANDS = {'fetch_artifacts', 'upload_all'}
if WHEELHOUSE_UPLOADER_COMMANDS.intersection(sys.argv):
import wheelhouse_uploader.cmd
cmdclass.update(vars(wheelhouse_uploader.cmd))

ext_modules =[
Extension(name="mlsauce.adaopt._adaoptc",
library_dirs=["mlsauce/adaopt/"],
runtime_library_dirs=["mlsauce/adaopt/"],
sources=["mlsauce/adaopt/_adaoptc.pyx"],
include_dirs=[numpy.get_include()]),
Extension(name="mlsauce.booster._boosterc",
library_dirs=["mlsauce/booster/"],
runtime_library_dirs=["mlsauce/booster/"],
sources=["mlsauce/booster/_boosterc.pyx"],
include_dirs=[numpy.get_include()]),
Extension(name="mlsauce.lasso._lassoc",
library_dirs=["mlsauce/lasso/"],
runtime_library_dirs=["mlsauce/lasso/"],
sources=["mlsauce/lasso/_lassoc.pyx"],
include_dirs=[numpy.get_include()]),
Extension(name="mlsauce.ridge._ridgec",
library_dirs=["mlsauce/ridge/"],
runtime_library_dirs=["mlsauce/ridge/"],
sources=["mlsauce/ridge/_ridgec.pyx"],
include_dirs=[numpy.get_include()]),
Extension(name="mlsauce.stump._stumpc",
library_dirs=["mlsauce/stump/"],
runtime_library_dirs=["mlsauce/stump/"],
sources=["mlsauce/stump/_stumpc.pyx"],
include_dirs=[numpy.get_include()]),
]

# Get the absolute path to the directory containing the setup script
abs_path = os.path.dirname(__file__)
rel_path = os.path.relpath(abs_path)
script_dir = rel_path
# Get absolute paths to Cython source files
adaopt_cython_file = str(script_dir + 'mlsauce/adaopt/_adaoptc.pyx')
booster_cython_file = str(script_dir + 'mlsauce/booster/_boosterc.pyx')
lasso_cython_file = str(script_dir + 'mlsauce/lasso/_lassoc.pyx')
ridge_cython_file = str(script_dir + 'mlsauce/ridge/_ridgec.pyx')
stump_cython_file = str(script_dir + 'mlsauce/stump/_stumpc.pyx')
# Update Extension definitions with absolute paths
ext_modules2 = [
Extension(name="mlsauce.adaopt._adaoptc",
sources=[adaopt_cython_file],
include_dirs=[numpy.get_include()]),
Extension(name="mlsauce.booster._boosterc",
sources=[booster_cython_file],
include_dirs=[numpy.get_include()]),
Extension(name="mlsauce.booster._lassoc",
sources=[booster_cython_file],
include_dirs=[numpy.get_include()]),
Extension(name="mlsauce.booster._ridgec",
sources=[booster_cython_file],
include_dirs=[numpy.get_include()]),
Extension(name="mlsauce.booster._stumpc",
sources=[booster_cython_file],
include_dirs=[numpy.get_include()]),
]


def setup_package():

install_all_requires = [
"numpy",
"Cython",
"joblib",
"matplotlib",
"pandas",
"requests",
"scikit-learn",
"scipy",
"tqdm"
]

if platform.system() in ('Linux', 'Darwin'):
install_jax_requires = ['jax', 'jaxlib']
else:
install_jax_requires = []

install_requires = [item for sublist in [install_all_requires, install_jax_requires] for item in sublist]

try:
cythonize_ext_modules = cythonize(ext_modules2)
except:
cythonize_ext_modules = cythonize(ext_modules)

metadata = dict(name=DISTNAME,
maintainer=MAINTAINER,
maintainer_email=MAINTAINER_EMAIL,
description=DESCRIPTION,
license=LICENSE,
version=VERSION,
long_description=LONG_DESCRIPTION,
classifiers=['Development Status :: 2 - Pre-Alpha',
'Intended Audience :: Developers',
'License :: OSI Approved :: BSD License',
'Natural Language :: English',
'Programming Language :: Python :: 3',
'Programming Language :: Python :: 3.5',
'Programming Language :: Python :: 3.6',
'Programming Language :: Python :: 3.7',
'Programming Language :: Python :: 3.8',
],
platforms=["linux", "macosx", "windows"],
python_requires=">=3.5",
install_requires=install_requires,
setup_requires=["numpy>= 1.13.0"],
package_data={'': ['*.pxd']},
packages=find_packages(),
ext_modules=cythonize_ext_modules,
**extra_setuptools_args)


setup(**metadata)

if __name__ == "__main__":
setup_package()
dir_path = os.path.dirname(os.path.realpath(__file__))
folders = ['adaopt', 'booster', 'elasticnet', 'lasso', 'ridge', 'stump']
for folder in folders:
filename = os.path.join(dir_path, "mlsauce", folder, 'setup2.py')
subprocess.run(['python3', filename, 'build_ext', '--inplace'])

0 comments on commit 7fd285f

Please sign in to comment.