Skip to content

Conversation

@nicoloboschi
Copy link
Contributor

No description provided.

# All query embeddings in the ColBERT documentation
# this name, EQ or Q, maps the exact name in most colBERT papers
Q = self.checkpoint.queryFromText(
queriesQ = self.checkpoint.queryFromText(
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Suggested change
queriesQ = self.checkpoint.queryFromText(
queries = self.checkpoint.queryFromText(
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Is there a way to align what's the variable in the paper and doc? I think the code should reflect what the paper describes. If Q cannot used, we should at least use queriesQ or something with a Q

Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Then just q lowercase ?

)

return Q
return queriesQ
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Suggested change
return queriesQ
return queries
@abstractmethod
def close(self):
"""Close the store."""
pass
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Suggested change
pass
self, query: str, k: Optional[int], query_maxlen: Optional[int], **kwargs
) -> List[Document]:
"""Retrieve documents from the store"""
pass
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Suggested change
pass
@cbornet
Copy link
Collaborator

cbornet commented Mar 19, 2024

Weird. Colbert is installed but the test complains that it doesn't find the module 😕

@nicoloboschi nicoloboschi merged commit 384b5a5 into main Mar 19, 2024
@nicoloboschi nicoloboschi deleted the colbert-cleanup branch March 19, 2024 12:44
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

4 participants