Exposed NumpyDType publicly (#791)

coreylowman · Jun 21, 2023 · 4748054 · 4748054
1 parent 3786aea
commit 4748054
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/tensor/mod.rs b/src/tensor/mod.rs
@@ -145,6 +145,8 @@ mod gradients;
 mod masks;
 #[cfg(feature = "numpy")]
 pub(crate) mod numpy;
+#[cfg(feature = "numpy")]
+pub use numpy::NumpyDtype;
 #[cfg(feature = "safetensors")]
 pub mod safetensors;
 mod tensorlike;