Mbonea commited on
Commit
963ef2b
1 Parent(s): 7450115

increase the size of the hits

Browse files
Files changed (1) hide show
  1. App/Embedding/utils/Elastic.py +1 -0
App/Embedding/utils/Elastic.py CHANGED
@@ -12,6 +12,7 @@ def FetchDocuments(ids, order="asc"):
12
  res = es.search(
13
  index=es_index_name,
14
  body={
 
15
  "query": {"terms": {"imdb_id.keyword": ids, "boost": 1.0}},
16
  "collapse": {
17
  "field": "imdb_id.keyword",
 
12
  res = es.search(
13
  index=es_index_name,
14
  body={
15
+ "size": 1000,
16
  "query": {"terms": {"imdb_id.keyword": ids, "boost": 1.0}},
17
  "collapse": {
18
  "field": "imdb_id.keyword",