tensor.nelements -> tensor.nelement

candle-pyo3/py_src/candle/__init__.pyi

@@ -344,7 +344,7 @@ class Tensor:
         """
         pass
     @property
-    def nelements(self) -> int:
+    def nelement(self) -> int:
         """
         Gets the tensor's element count.
         """

candle-pyo3/py_src/candle/testing/__init__.py

@@ -67,4 +67,4 @@ def assert_almost_equal(
 
     threshold = (expected.abs().to_dtype(candle.f32) * rtol + atol).to(expected)
 
-    assert (diff <= threshold).sum_all().values() == actual.nelements, f"Difference between tensors was to great"
+    assert (diff <= threshold).sum_all().values() == actual.nelement, f"Difference between tensors was to great"

candle-pyo3/src/lib.rs

@@ -337,7 +337,7 @@ impl PyTensor {
     #[getter]
     /// Gets the tensor's element count.
     /// &RETURNS&: int
-    fn nelements(&self) -> usize {
+    fn nelement(&self) -> usize {
         self.0.elem_count()
     }