Add symlink to setup.py instead of hnswlib

taoseng · Dec 8, 2020 · 6efa48c · 6efa48c
1 parent afd18d2
commit 6efa48c
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 3 deletions.
diff --git a/python_bindings/hnswlib b/python_bindings/hnswlib
diff --git a/python_bindings/setup.py b/python_bindings/setup.py
@@ -0,0 +1 @@
+../setup.py
diff --git a/setup.py b/setup.py
@@ -6,9 +6,15 @@
 
 __version__ = '0.4.0'
 
+# compatibility when run in python_bindings
+bindings_dir = 'python_bindings'
+if bindings_dir in os.path.basename(os.getcwd()):
+    source_files = ['./bindings.cpp']
+    include_dirs = ['../hnswlib/']
+else:
+    source_files = ['./python_bindings/bindings.cpp']
+    include_dirs = ['./hnswlib/']
 
-source_files = ['./python_bindings/bindings.cpp']
-include_dirs = ['./hnswlib/']
 
 libraries = []
 extra_objects = []