geekyrakshit commited on
Commit
c063304
·
1 Parent(s): 9595675

upfate: CalPaliRetriever methods to align with other retrievers

Browse files
medrag_multi_modal/retrieval/colpali_retrieval.py CHANGED
@@ -59,7 +59,7 @@ class CalPaliRetriever(weave.Model):
59
  from medrag_multi_modal.retrieval import CalPaliRetriever
60
 
61
  weave.init(project_name="ml-colabs/medrag-multi-modal")
62
- retriever = CalPaliRetriever.from_artifact(
63
  index_artifact_name="ml-colabs/medrag-multi-modal/grays-anatomy:v0",
64
  metadata_dataset_name="grays-anatomy-images:v0",
65
  data_artifact_name="ml-colabs/medrag-multi-modal/grays-anatomy-images:v1",
@@ -100,7 +100,7 @@ class CalPaliRetriever(weave.Model):
100
  )
101
 
102
  @classmethod
103
- def from_artifact(
104
  cls,
105
  index_artifact_name: str,
106
  metadata_dataset_name: str,
 
59
  from medrag_multi_modal.retrieval import CalPaliRetriever
60
 
61
  weave.init(project_name="ml-colabs/medrag-multi-modal")
62
+ retriever = CalPaliRetriever.from_wandb_artifact(
63
  index_artifact_name="ml-colabs/medrag-multi-modal/grays-anatomy:v0",
64
  metadata_dataset_name="grays-anatomy-images:v0",
65
  data_artifact_name="ml-colabs/medrag-multi-modal/grays-anatomy-images:v1",
 
100
  )
101
 
102
  @classmethod
103
+ def from_wandb_artifact(
104
  cls,
105
  index_artifact_name: str,
106
  metadata_dataset_name: str,