Skip to content

Commit

Permalink
Merge pull request #22 from Techtonique/cythonize
Browse files Browse the repository at this point in the history
Cythonize
  • Loading branch information
thierrymoudiki authored Jan 31, 2024
2 parents dc70991 + b4ea185 commit 1bdb272
Showing 1 changed file with 18 additions and 16 deletions.
34 changes: 18 additions & 16 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,13 +5,14 @@

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

from distutils.command.clean import clean as Clean
from distutils.core import Extension, setup

from os import path
from setuptools import find_packages

try:
Expand All @@ -36,7 +37,7 @@
MAINTAINER_EMAIL = '[email protected]'
LICENSE = 'BSD3 Clause Clear'

__version__ = '0.10.0'
__version__ = '0.10.1'

VERSION = __version__

Expand Down Expand Up @@ -143,21 +144,22 @@ def run(self):

def setup_package():

# get the dependencies and installs
here = path.abspath(path.dirname(__file__))

with open(path.join(here, 'requirements.txt'), encoding='utf-8') as f:
all_reqs = f.read().split('\n')

install_all_requires = [
'numpy>={}'.format(NUMPY_MIN_VERSION),
'scipy>={}'.format(SCIPY_MIN_VERSION),
'joblib>={}'.format(JOBLIB_MIN_VERSION),
'scikit-learn>={}'.format(SKLEARN_MIN_VERSION),
'pandas>={}'.format(PANDAS_MIN_VERSION),
'requests>={}'.format(REQUESTS_MIN_VERSION),
]
install_jax_requires = [
'jax',
'jaxlib'
] if platform.system() in ('Linux', 'Darwin') else []
other_requirements = ["tqdm"]

install_requires = [item for sublist in [install_jax_requires, install_all_requires, other_requirements] for item in sublist]
x.strip() for x in all_reqs if "git+" not in x
]

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]

metadata = dict(name=DISTNAME,
maintainer=MAINTAINER,
Expand Down

0 comments on commit 1bdb272

Please sign in to comment.