fix scikit bug on mac os

Xtra-Computing · Jun 10, 2018 · 3ebdf43 · 3ebdf43
1 parent 9963067
commit 3ebdf43
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/python/thundersvmScikit.py b/python/thundersvmScikit.py
@@ -20,8 +20,8 @@
     lib_path = path.join(dirname, '../build/lib/libthundersvm.so')
 elif platform == "win32":
     lib_path = path.join(dirname, '../build/bin/Debug/thundersvm.dll')
-# elif platform == "darwin":
-#     lib_path = path.join(dirname, '../build/lib/libthundersvm.dylib')
+elif platform == "darwin":
+    lib_path = path.join(dirname, '../build/lib/libthundersvm.dylib')
 else :
     print ("OS not supported!")
     exit()
@@ -56,6 +56,7 @@ def __init__(self, kernel, degree,
         self.n_jobs = n_jobs
         self.random_state = random_state
         self.max_mem_size = max_mem_size
+        thundersvm.model_new.restype = c_void_p
         self.model = thundersvm.model_new(SVM_TYPE.index(self._impl))
 
     def label_validate(self, y):