add allowed

Signed-off-by: Andrew Duffy <andrew@a10y.dev>

Author: a10y

encodings/fastlanes/src/lib.rs

@@ -35,7 +35,7 @@ use vortex_array::session::ArraySessionExt;
 use vortex_session::VortexSession;
 
 /// Check if we're using experimental patches deserialization
-static USE_EXPERIMENTAL_PATCHES: LazyLock<bool> =
+pub static USE_EXPERIMENTAL_PATCHES: LazyLock<bool> =
     LazyLock::new(|| env::var("VORTEX_EXPERIMENTAL_PATCHED_ARRAY").is_ok());
 
 /// Initialize fastlanes encodings in the given session.

vortex-btrblocks/src/schemes/integer.rs

@@ -3,9 +3,6 @@
 
 //! Integer compression schemes.
 
-use std::env;
-use std::sync::LazyLock;
-
 use vortex_array::ArrayRef;
 use vortex_array::Canonical;
 use vortex_array::IntoArray;
@@ -35,6 +32,7 @@ use vortex_fastlanes::FoR;
 use vortex_fastlanes::FoRArrayExt;
 use vortex_fastlanes::RLE;
 use vortex_fastlanes::RLEArrayExt;
+use vortex_fastlanes::USE_EXPERIMENTAL_PATCHES;
 use vortex_fastlanes::bitpack_compress::bit_width_histogram;
 use vortex_fastlanes::bitpack_compress::bitpack_encode;
 use vortex_fastlanes::bitpack_compress::find_best_bit_width;
@@ -298,10 +296,6 @@ impl Scheme for ZigZagScheme {
     }
 }
 
-// replicated from vortex-file
-static USE_EXPERIMENTAL_PATCHES: LazyLock<bool> =
-    LazyLock::new(|| env::var("VORTEX_EXPERIMENTAL_PATCHED_ARRAY").is_ok());
-
 impl Scheme for BitPackingScheme {
     fn scheme_name(&self) -> &'static str {
         "vortex.int.bitpacking"

vortex-file/src/strategy.rs

@@ -21,6 +21,7 @@ use vortex_array::arrays::List;
 use vortex_array::arrays::ListView;
 use vortex_array::arrays::Masked;
 use vortex_array::arrays::Null;
+use vortex_array::arrays::Patched;
 use vortex_array::arrays::Primitive;
 use vortex_array::arrays::Struct;
 use vortex_array::arrays::VarBin;
@@ -90,6 +91,10 @@ pub static ALLOWED_ENCODINGS: LazyLock<HashSet<ArrayId>> = LazyLock::new(|| {
     allowed.insert(Masked.id());
     allowed.insert(Dict.id());
 
+    if *vortex_fastlanes::USE_EXPERIMENTAL_PATCHES {
+        allowed.insert(Patched.id());
+    }
+
     // Compressed encodings from encoding crates
     allowed.insert(ALP.id());
     allowed.insert(ALPRD.id());