expose tensorops and error to root

kornia · Oct 8, 2024 · b9e1173 · b9e1173
1 parent c847a8b
commit b9e1173
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/crates/kornia-core-ops/src/lib.rs b/crates/kornia-core-ops/src/lib.rs
@@ -6,3 +6,6 @@ pub mod error;
 
 /// module containing ops implementations.
 pub mod ops;
+
+pub use crate::error::TensorOpsError;
+pub use crate::ops::TensorOps;
diff --git a/crates/kornia-core-ops/src/ops.rs b/crates/kornia-core-ops/src/ops.rs
@@ -49,7 +49,7 @@ where
     ///
     /// ```
     /// use kornia_core::{Tensor, CpuAllocator};
-    /// use kornia_core_ops::ops::TensorOps;
+    /// use kornia_core_ops::TensorOps;
     ///
     /// let data: [u8; 6] = [1, 1, 1, 1, 1, 1];
     /// let t = Tensor::<u8, 2>::from_shape_slice([2, 3], &data, CpuAllocator).unwrap();
@@ -80,7 +80,7 @@ where
     ///
     /// ```
     /// use kornia_core::{Tensor, CpuAllocator};
-    /// use kornia_core_ops::ops::TensorOps;
+    /// use kornia_core_ops::TensorOps;
     ///
     /// let data: [f32; 5] = [3., 3., 2., 1., 1.];
     /// let t = Tensor::<f32, 1>::from_shape_slice([5], &data, CpuAllocator).unwrap();
@@ -110,7 +110,7 @@ where
     ///
     /// ```
     /// use kornia_core::{Tensor, CpuAllocator};
-    /// use kornia_core_ops::ops::TensorOps;
+    /// use kornia_core_ops::TensorOps;
     ///
     /// let data: [f32; 5] = [3., 2., 1., 1., 1.];
     /// let t = Tensor::<f32, 1>::from_shape_slice([5], &data, CpuAllocator).unwrap();