@@ -8,7 +8,7 @@ use bytes::Bytes;
88use pyo3:: exceptions:: PyValueError ;
99use pyo3:: prelude:: * ;
1010use pyo3:: { Bound , Python } ;
11- use vortex:: ArrayRegistry ;
11+ use vortex:: file :: DEFAULT_REGISTRY ;
1212use vortex_ipc:: messages:: { DecoderMessage , MessageDecoder , PollRead } ;
1313
1414use crate :: arrays:: PyArrayRef ;
@@ -48,7 +48,7 @@ pub(crate) fn init(py: Python, parent: &Bound<PyModule>) -> PyResult<()> {
4848/// The decoded Vortex array
4949#[ pyfunction]
5050fn 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