Fix Mac OSX build (#56)

Summary:
Properly determine if we are on Mac or now and add linker flags
Pull Request resolved: #56

Differential Revision: D19330182

Pulled By: lematt1991

fbshipit-source-id: f9651a47d8981a9341f4f19c30663cb5462d114b

setup.py

@@ -11,28 +11,32 @@
 from distutils.extension import Extension
 from subprocess import check_output
 from distutils import sysconfig
-import re
+import sys
 
 extra_compile_args = ['-std=c++11']
+extra_link_args = []
 
 # Super hacky way of determining if clang or gcc is being used
 CC = sysconfig.get_config_vars().get('CC', 'gcc').split(' ')[0]
 out = check_output([CC, '--version'])
-if re.search('apple *llvm', str(out.lower())):
-    extra_compile_args.append('-stdlib=libc++')
+if sys.platform == 'darwin':
+    extra_compile_args = ["-stdlib=libc++"]
+    extra_link_args=['-stdlib=libc++']
 
 extensions = [
     Extension(
         "hype.graph_dataset",
         ["hype/graph_dataset.pyx"],
         include_dirs=[numpy.get_include()],
+        extra_link_args=extra_link_args,
         extra_compile_args=extra_compile_args,
         language='c++',
     ),
     Extension(
         "hype.adjacency_matrix_dataset",
         ["hype/adjacency_matrix_dataset.pyx"],
         include_dirs=[numpy.get_include()],
+        extra_link_args=extra_link_args,
         extra_compile_args=extra_compile_args,
         language='c++',
     ),