sij/khoj
1
0
Fork 0
mirror of https://github.com/khoj-ai/khoj.git synced 2024-11-27 17:35:07 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

Simplify extracting entries satisfying file filter

Browse source
This commit is contained in:
Debanjum Singh Solanky 2022-09-05 02:09:36 +03:00
parent 7606724dbc
commit 2890b4cd44
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
src/search_filter/file_filter.py
View file

@ -67,7 +67,7 @@ class FileFilter(BaseFilter):
# Get entries (and associated embeddings) satisfying file filters # Get entries (and associated embeddings) satisfying file filters
start = time.time() start = time.time()
entries = [entry for id, entry in enumerate(raw_entries) if id in included_entry_indices] entries = [raw_entries[id] for id in included_entry_indices]
embeddings = torch.index_select(raw_embeddings, 0, torch.tensor(list(included_entry_indices))) embeddings = torch.index_select(raw_embeddings, 0, torch.tensor(list(included_entry_indices)))
end = time.time() end = time.time()