Karthikeyen92 commited on
Commit
5761cc2
·
verified ·
1 Parent(s): 7d8ba03

Update py/db_storage.py

Browse files
Files changed (1) hide show
  1. py/db_storage.py +6 -2
py/db_storage.py CHANGED
@@ -164,9 +164,13 @@ class DBStorage:
164
  check_and_delete(FAISS_DB_PATH)
165
  vector_store.save_local(FAISS_DB_PATH)
166
 
167
- def check_and_delete(PATH):
168
  if os.path.isdir(PATH):
169
- shutil.rmtree(PATH, onexc=lambda func, path, exc: os.chmod(path, 0o777))
 
 
 
 
170
  print(f'Deleted {PATH}')
171
 
172
 
 
164
  check_and_delete(FAISS_DB_PATH)
165
  vector_store.save_local(FAISS_DB_PATH)
166
 
167
+ def check_and_delete(PATH, hugg=False):
168
  if os.path.isdir(PATH):
169
+ if hugg:
170
+ shutil.rmtree(PATH)
171
+ else:
172
+ shutil.rmtree(PATH, onexc=lambda func, path, exc: os.chmod(path, 0o777))
173
+
174
  print(f'Deleted {PATH}')
175
 
176