removed extra_link_args for macOS (#10)

* print the system platform

* print out the platform

* changed darwin to Darwin

* remove extra_link_args for mac

Author: LeilyR

setup.py

@@ -18,13 +18,21 @@ def get_include():  # TODO
 def __extra_compile_args():
     extra_compile_args = []
     print("Platform is "+ platform.system())
-    if platform.system() == 'darwin' or 'macOS':
+
+    if platform.system() == 'Darwin':
         extra_compile_args = ["-std=c++11"]
     else:
         extra_compile_args = ["-fopenmp", "-std=c++11"]
     return extra_compile_args
 
 
+def __extra_link_args():
+    extra_link_args = []
+    if platform.system() != 'Darwin':
+        extra_link_args = ["-lgomp", "-lm", "-lrt"]
+    return extra_link_args
+
+
 sources_list = ['src/krbalancing.cpp']
 
 kr_module = Extension('krbalancing',
@@ -33,7 +41,7 @@ def __extra_compile_args():
                           # Path to eigen3 headers
                           get_include()
                       ],
-                      extra_link_args=["-lgomp", "-lm", "-lrt"],
+                      extra_link_args=__extra_link_args(),
                       extra_compile_args=__extra_compile_args()
                       )