revert page column (#8217)

This commit is contained in:
Jyong 2024-09-10 18:21:22 +08:00 committed by GitHub
commit bb3002b173
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 1 additions and 6 deletions

View file

@ -27,7 +27,7 @@ class AbstractVectorFactory(ABC):
class Vector:
def __init__(self, dataset: Dataset, attributes: list = None):
if attributes is None:
attributes = ["doc_id", "dataset_id", "document_id", "doc_hash", "page"]
attributes = ["doc_id", "dataset_id", "document_id", "doc_hash"]
self._dataset = dataset
self._embeddings = self._get_embeddings()
self._attributes = attributes