diff --git a/docarray/array/storage/sqlite/backend.py b/docarray/array/storage/sqlite/backend.py index 4332467c964..fe9627be194 100644 --- a/docarray/array/storage/sqlite/backend.py +++ b/docarray/array/storage/sqlite/backend.py @@ -34,7 +34,7 @@ def _sanitize_table_name(table_name: str) -> str: class SqliteConfig: connection: Optional[Union[str, 'sqlite3.Connection']] = None table_name: Optional[str] = None - serialize_config: Dict = field(default_factory=lambda: {'protocol': 'protobuf'}) + serialize_config: Dict = field(default_factory=dict) conn_config: Dict = field(default_factory=dict) journal_mode: str = 'DELETE' synchronous: str = 'OFF' diff --git a/docarray/array/storage/weaviate/backend.py b/docarray/array/storage/weaviate/backend.py index 07a0e642636..df08cef46d7 100644 --- a/docarray/array/storage/weaviate/backend.py +++ b/docarray/array/storage/weaviate/backend.py @@ -33,7 +33,7 @@ class WeaviateConfig: client: Optional[Union[str, weaviate.Client]] = None n_dim: Optional[int] = None name: Optional[str] = None - serialize_config: Dict = field(default_factory=lambda: {'protocol': 'protobuf'}) + serialize_config: Dict = field(default_factory=dict) class BackendMixin(BaseBackendMixin): diff --git a/tests/unit/array/mixins/test_embed.py b/tests/unit/array/mixins/test_embed.py index b453fc39c9d..0ae985e1650 100644 --- a/tests/unit/array/mixins/test_embed.py +++ b/tests/unit/array/mixins/test_embed.py @@ -43,7 +43,7 @@ ) -@pytest.mark.parametrize('framework', ['onnx', 'keras', 'pytorch', 'paddle']) +@pytest.mark.parametrize('framework', ['onnx', 'keras', 'pytorch']) @pytest.mark.parametrize( 'da', [DocumentArray, DocumentArraySqlite, DocumentArrayWeaviate] )