Update setup.py - - find pyx

Techtonique · Apr 23, 2024 · 025b341 · 025b341
1 parent cfcc72a
commit 025b341
Showing 1 changed file with 6 additions and 7 deletions.
diff --git a/setup.py b/setup.py
@@ -144,13 +144,12 @@ def run(self):
 ]
 
 # Get the absolute path to the directory containing the setup script
-script_dir = os.path.basename(__file__)
-# 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')
+script_dir = os.path.dirname(os.path.realpath(__file__))
+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",