alfraser commited on
Commit
c2d8e03
1 Parent(s): cc94398

Fixed bug aligning the names of the vector stores on train versus inference

Browse files
Files changed (1) hide show
  1. src/training/train_rag.py +1 -1
src/training/train_rag.py CHANGED
@@ -60,7 +60,7 @@ def connect_to_product_db(args) -> None:
60
 
61
 
62
  def get_out_vector_store_client(args, allow_reset: bool = False) -> chromadb.Client:
63
- out_dir = os.path.join(data_dir, 'vector_stores', args.out_vectors)
64
  chroma_settings = Settings()
65
  chroma_settings.allow_reset = allow_reset
66
  return chromadb.PersistentClient(path=out_dir, settings=chroma_settings)
 
60
 
61
 
62
  def get_out_vector_store_client(args, allow_reset: bool = False) -> chromadb.Client:
63
+ out_dir = os.path.join(data_dir, 'vector_stores', f"{args.out_vectors}_chroma")
64
  chroma_settings = Settings()
65
  chroma_settings.allow_reset = allow_reset
66
  return chromadb.PersistentClient(path=out_dir, settings=chroma_settings)