From 94fac1076addd29f0ddcd411f4c14a6153a78122 Mon Sep 17 00:00:00 2001 From: Tianqi Chen Date: Sun, 26 Apr 2015 00:17:58 -0700 Subject: [PATCH] bugfix setup --- wrapper/setup.py | 5 +++-- wrapper/xgboost.py | 5 ++--- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/wrapper/setup.py b/wrapper/setup.py index bcf3e7ea8..14f9e9b55 100644 --- a/wrapper/setup.py +++ b/wrapper/setup.py @@ -12,9 +12,10 @@ dll_path = [curr_dir] if os.name == 'nt': if platform.architecture()[0] == '64bit': - dll_path.append(os.path.join(curr_dir, '../windows/Release/')) - else: dll_path.append(os.path.join(curr_dir, '../windows/x64/Release/')) + else: + dll_path.append(os.path.join(curr_dir, '../windows/Release/')) + if os.name == 'nt': dll_path = [os.path.join(p, 'xgboost_wrapper.dll') for p in dll_path] diff --git a/wrapper/xgboost.py b/wrapper/xgboost.py index 8cee737e9..d9bcecd70 100644 --- a/wrapper/xgboost.py +++ b/wrapper/xgboost.py @@ -43,10 +43,9 @@ def load_xglib(): dll_path = [curr_path] if os.name == 'nt': if platform.architecture()[0] == '64bit': - dll_path.append(os.path.join(curr_dir, '../windows/Release/')) + dll_path.append(os.path.join(curr_path, '../windows/x64/Release/')) else: - dll_path.append(os.path.join(curr_dir, '../windows/x64/Release/')) - + dll_path.append(os.path.join(curr_path, '../windows/Release/')) if os.name == 'nt': dll_path = [os.path.join(p, 'xgboost_wrapper.dll') for p in dll_path] else: