vikramvasudevan commited on
Commit
0412bab
·
verified ·
1 Parent(s): 3772fe4

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. config.py +9 -1
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