Heng2004 commited on
Commit
b4ddaff
Β·
verified Β·
1 Parent(s): c053894

Update qa_store.py

Browse files
Files changed (1) hide show
  1. qa_store.py +3 -3
qa_store.py CHANGED
@@ -22,11 +22,11 @@ NEXT_MANUAL_ID: int = 1
22
 
23
  # Embeddings for textbook entries (one vector per ENTRIES item)
24
  # Will be set to a torch.Tensor by _build_entry_embeddings() in model_utils.py
25
- TEXT_EMBEDDINGS = None
26
 
27
  # πŸ‘‡ Add these for glossary
28
- GLOSSARY = [] # list of glossary items (dicts)
29
- GLOSSARY_EMBEDDINGS = None # numpy array of embeddings, same size as GLOSSARY
30
 
31
 
32
  def normalize_question(q: str) -> str:
 
22
 
23
  # Embeddings for textbook entries (one vector per ENTRIES item)
24
  # Will be set to a torch.Tensor by _build_entry_embeddings() in model_utils.py
25
+ TEXT_EMBEDDINGS = None # torch.Tensor
26
 
27
  # πŸ‘‡ Add these for glossary
28
+ GLOSSARY: List[Dict[str, Any]] = [] # list of glossary items (dicts)
29
+ GLOSSARY_EMBEDDINGS = None # numpy.ndarray
30
 
31
 
32
  def normalize_question(q: str) -> str: