Fixed the time.time bug here. Also a call to reset the Chroma DB

src/training/train_rag.py

@@ -71,7 +71,7 @@ def prepare_to_vectorise(args) -> chromadb.Client:
 
     # Now do possibly destructive setup
     if args.overwrite:
-        empty_or_create_out_vector_store()
+        empty_or_create_out_vector_store(args)
     elif need_to_copy_vector_store(args):
         copy_vector_store(args)
 
@@ -132,7 +132,7 @@ def train(args):
 
 
 if __name__ == "__main__":
-    start = time.time()
+    start = time()
     train(args)
-    end = time.time()
+    end = time()
     print(f"Training took {end-start:.2f} seconds")