mirror of
https://github.com/wassname/stampy-chat.git
synced 2026-09-12 13:00:42 +08:00
Removed time logging and show_stats
This commit is contained in:
@@ -1,6 +1,5 @@
|
||||
# dataset/update_dataset.py
|
||||
|
||||
import time
|
||||
from typing import Dict, List
|
||||
import numpy as np
|
||||
import logging
|
||||
@@ -66,7 +65,6 @@ class ARDUpdater:
|
||||
iterable_data = iterable_data.filter(lambda entry: self.db.upsert_entry(entry))
|
||||
|
||||
for entry in tqdm(iterable_data):
|
||||
t_entry_start = time.time()
|
||||
try:
|
||||
self.pinecone_handler.delete_entry(entry['id'])
|
||||
|
||||
@@ -79,9 +77,6 @@ class ARDUpdater:
|
||||
except Exception as e:
|
||||
self.logger.error(f"An error occurred while updating source {source}: {str(e)}", exc_info=True)
|
||||
|
||||
t_entry_end = time.time()
|
||||
self.logger.info(f"Time for processing one entry: {t_entry_end - t_entry_start} seconds")
|
||||
|
||||
self.logger.info(f"Successfully updated {source} entries.")
|
||||
|
||||
def preprocess(self, entry):
|
||||
@@ -132,11 +127,6 @@ class ARDUpdater:
|
||||
|
||||
return embeddings
|
||||
|
||||
def show_stats(self): #TODO
|
||||
# Show index
|
||||
# Show database
|
||||
pass
|
||||
|
||||
|
||||
##### Helper functions #####
|
||||
|
||||
|
||||
Reference in New Issue
Block a user