Update __init__.py

Allow for flexible path setting
AmenRa · Sep 5, 2023 · 6f9f099 · 6f9f099
1 parent 1a4e3ff
commit 6f9f099
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