fixed variable naming

databio · Nov 13, 2023 · 12d94db · 12d94db
1 parent 390e791
commit 12d94db
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/bbconf/bbconf.py b/bbconf/bbconf.py
@@ -17,7 +17,7 @@
     CFG_PATH_KEY,
     CFG_PATH_PIPELINE_OUTPUT_KEY,
     CFG_PATH_BEDSTAT_DIR_KEY,
-    CFG_PATH_SENTENCE2VEC_KEY,
+    CFG_PATH_TEXT2VEC_KEY,
     DEFAULT_SECTION_VALUES,
     CFG_PATH_BEDBUNCHER_DIR_KEY,
     BED_TABLE,
@@ -35,7 +35,7 @@
     CFG_QDRANT_API_KEY,
     CFG_QDRANT_HOST_KEY,
     CFG_QDRANT_COLLECTION_NAME_KEY,
-    DEFAULT_SENTENCE2VEC_MODEL,
+    DEFAULT_TEXT2VEC_MODEL,
     DEFAULT_VEC2VEC_MODEL,
     DEFAULT_REGION2_VEC_MODEL,
     CFG_ACCESS_METHOD_KEY,
@@ -103,7 +103,7 @@ def __init__(self, config_path: str = None, database_only: bool = False):
         self._t2bsi = None
         try:
             self._senta2vec_hg_model_name = self.config[CFG_PATH_KEY].get(
-                CFG_PATH_SENTENCE2VEC_KEY, DEFAULT_SENTENCE2VEC_MODEL
+                CFG_PATH_TEXT2VEC_KEY, DEFAULT_TEXT2VEC_MODEL
             )
             _LOGGER.debug("Setting up qdrant database connection...")
             if self.config[CFG_QDRANT_KEY].get(CFG_QDRANT_API_KEY, None):

diff --git a/bbconf/const.py b/bbconf/const.py
@@ -35,7 +35,7 @@
 CFG_PATH_PIPELINE_OUTPUT_KEY = "pipeline_output_path"
 CFG_PATH_REGION2VEC_KEY = "region2vec"
 CFG_PATH_VEC2VEC_KEY = "vec2vec"
-CFG_PATH_SENTENCE2VEC_KEY = "sentence2vec"
+CFG_PATH_TEXT2VEC_KEY = "text2vec"
 
 
 CFG_DATABASE_KEY = "database"
@@ -95,7 +95,7 @@
     },
 }
 
-DEFAULT_SENTENCE2VEC_MODEL = "sentence-transformers/all-MiniLM-L6-v2"
+DEFAULT_TEXT2VEC_MODEL = "sentence-transformers/all-MiniLM-L6-v2"
 DEFAULT_VEC2VEC_MODEL = "databio/v2v-MiniLM-v2-ATAC-hg38"
 DEFAULT_REGION2_VEC_MODEL = "databio/r2v-ChIP-atlas-hg38"