Skip to content

Commit 7492524

Browse files
committed
use default registry
Signed-off-by: Onur Satici <[email protected]>
1 parent f76b30a commit 7492524

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

vortex-python/src/serde/mod.rs

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@ use bytes::Bytes;
88
use pyo3::exceptions::PyValueError;
99
use pyo3::prelude::*;
1010
use pyo3::{Bound, Python};
11-
use vortex::ArrayRegistry;
11+
use vortex::file::DEFAULT_REGISTRY;
1212
use vortex_ipc::messages::{DecoderMessage, MessageDecoder, PollRead};
1313

1414
use crate::arrays::PyArrayRef;
@@ -48,7 +48,7 @@ pub(crate) fn init(py: Python, parent: &Bound<PyModule>) -> PyResult<()> {
4848
/// The decoded Vortex array
4949
#[pyfunction]
5050
fn decode_ipc_array(array_bytes: Vec<u8>, dtype_bytes: Vec<u8>) -> PyResult<PyArrayRef> {
51-
let registry = ArrayRegistry::canonical_only();
51+
let registry = DEFAULT_REGISTRY.as_ref().clone();
5252
let mut decoder = MessageDecoder::new(registry);
5353

5454
let mut dtype_buf = Bytes::from(dtype_bytes);
@@ -102,7 +102,7 @@ fn decode_ipc_array_buffers<'py>(
102102
) -> PyResult<PyArrayRef> {
103103
use pyo3::buffer::PyBuffer;
104104

105-
let registry = ArrayRegistry::canonical_only();
105+
let registry = DEFAULT_REGISTRY.as_ref().clone();
106106
let mut decoder = MessageDecoder::new(registry);
107107

108108
// Concatenate dtype buffers

0 commit comments

Comments
 (0)