alfraser commited on
Commit
cc94398
·
1 Parent(s): d7a31e7

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 +2 -2
src/training/train_rag.py CHANGED
@@ -33,8 +33,8 @@ def need_to_copy_vector_store(args) -> bool:
33
 
34
 
35
  def copy_vector_store(args) -> None:
36
- src = os.path.join(data_dir, 'vector_stores', args.in_vectors)
37
- dest = os.path.join(data_dir, 'vector_stores', args.out_vectors)
38
  shutil.copytree(src, dest)
39
 
40
 
 
33
 
34
 
35
  def copy_vector_store(args) -> None:
36
+ src = os.path.join(data_dir, 'vector_stores', f"{args.in_vectors}_chroma")
37
+ dest = os.path.join(data_dir, 'vector_stores', f"{args.out_vectors}_chroma")
38
  shutil.copytree(src, dest)
39
 
40