Merge pull request #26 from alex2awesome/patch-1

AmenRa · Sep 5, 2023 · 3939729 · 3939729
2 parents 1a4e3ff + 6f9f099
commit 3939729
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/retriv/__init__.py b/retriv/__init__.py
@@ -21,8 +21,8 @@
 
 # Set environment variables ----------------------------------------------------
 os.environ["TOKENIZERS_PARALLELISM"] = "false"
-os.environ["RETRIV_BASE_PATH"] = str(Path.home() / ".retriv")
-
+if "RETRIV_BASE_PATH" not in os.environ: # allow user to set a different path in .bash_profile
+    os.environ["RETRIV_BASE_PATH"] = str(Path.home() / ".retriv")
 
 def set_base_path(path: str):
     os.environ["RETRIV_BASE_PATH"] = path