Expose the deltalake-aws crate from the meta crate when present

delta-io · Dec 30, 2023 · 00c56ed · 00c56ed
1 parent eb898bf
commit 00c56ed
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/crates/deltalake/src/lib.rs b/crates/deltalake/src/lib.rs
@@ -2,3 +2,6 @@
  * The deltalake crate is currently just a meta-package shim for deltalake-core
  */
 pub use deltalake_core::*;
+
+#[cfg(any(feature="s3", feature="s3-native-tls"))]
+pub use deltalake_aws as aws;
diff --git a/python/src/lib.rs b/python/src/lib.rs
@@ -1401,6 +1401,8 @@ impl PyDeltaDataChecker {
 // module name need to match project name
 fn _internal(py: Python, m: &PyModule) -> PyResult<()> {
     use crate::error::{CommitFailedError, DeltaError, TableNotFoundError};
+
+    deltalake::aws::register_handlers(None);
     m.add("DeltaError", py.get_type::<DeltaError>())?;
     m.add("CommitFailedError", py.get_type::<CommitFailedError>())?;
     m.add("DeltaProtocolError", py.get_type::<DeltaProtocolError>())?;