roni commited on
Commit
53aa3a7
·
1 Parent(s): 1694358

filtering on indexes with data

Browse files
Files changed (1) hide show
  1. get_index.py +3 -1
get_index.py CHANGED
@@ -25,5 +25,7 @@ def get_engines(index_repo: str, model_repo: str):
25
  engines = {}
26
  for subindex_path in subindex_paths:
27
  subindex_name = os.path.basename(subindex_path)
28
- engines[subindex_name] = ProteinSearchEngine(data_path=Path(subindex_path))
 
 
29
  return engines
 
25
  engines = {}
26
  for subindex_path in subindex_paths:
27
  subindex_name = os.path.basename(subindex_path)
28
+ engine = ProteinSearchEngine(data_path=Path(subindex_path))
29
+ if len(engine) > 10000:
30
+ engines[subindex_name] = engine
31
  return engines