Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Upload folder using huggingface_hub
Browse files
config.py
CHANGED
|
@@ -766,6 +766,14 @@ class SanatanConfig:
|
|
| 766 |
embedding_fn = scripture["collection_embedding_fn"] # overridden in config
|
| 767 |
return embedding_fn
|
| 768 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 769 |
def filter_scriptures_fields(self, fields_to_keep: List[str]) -> List[Dict]:
|
| 770 |
"""
|
| 771 |
Return a list of scripture dicts containing only the specified fields.
|
|
@@ -773,7 +781,7 @@ class SanatanConfig:
|
|
| 773 |
filtered = []
|
| 774 |
for s in self.scriptures:
|
| 775 |
filtered.append({k: s[k] for k in fields_to_keep if k in s})
|
| 776 |
-
return filtered
|
| 777 |
|
| 778 |
def canonicalize_document(
|
| 779 |
self, scripture_name: str, document_text: str, metadata_doc: dict
|
|
|
|
| 766 |
embedding_fn = scripture["collection_embedding_fn"] # overridden in config
|
| 767 |
return embedding_fn
|
| 768 |
|
| 769 |
+
def remove_callables(self,obj):
|
| 770 |
+
if isinstance(obj, dict):
|
| 771 |
+
return {k: self.remove_callables(v) for k, v in obj.items() if not callable(v)}
|
| 772 |
+
elif isinstance(obj, list):
|
| 773 |
+
return [self.remove_callables(v) for v in obj if not callable(v)]
|
| 774 |
+
else:
|
| 775 |
+
return obj
|
| 776 |
+
|
| 777 |
def filter_scriptures_fields(self, fields_to_keep: List[str]) -> List[Dict]:
|
| 778 |
"""
|
| 779 |
Return a list of scripture dicts containing only the specified fields.
|
|
|
|
| 781 |
filtered = []
|
| 782 |
for s in self.scriptures:
|
| 783 |
filtered.append({k: s[k] for k in fields_to_keep if k in s})
|
| 784 |
+
return self.remove_callables(filtered)
|
| 785 |
|
| 786 |
def canonicalize_document(
|
| 787 |
self, scripture_name: str, document_text: str, metadata_doc: dict
|