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: