Update qa_store.py
Browse files- 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
|
| 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:
|