From 54621d46345bb9c460f2d528ef66056bb6ebd53c Mon Sep 17 00:00:00 2001
From: NAUD Maxence <maxence.naud@cea.fr>
Date: Mon, 20 Jan 2025 02:53:10 +0000
Subject: [PATCH] Add offset to Python format of Tensor
---
python_binding/data/pybind_Tensor.cpp | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/python_binding/data/pybind_Tensor.cpp b/python_binding/data/pybind_Tensor.cpp
index 973fc6f9a..2171d4897 100644
--- a/python_binding/data/pybind_Tensor.cpp
+++ b/python_binding/data/pybind_Tensor.cpp
@@ -345,7 +345,7 @@ void init_Tensor(py::module& m){
return b.toString();
})
.def("__repr__", [](Tensor& b) {
- return fmt::format("Tensor(dims = {}, dtype = {})", b.dims(), std::string(EnumStrings<DataType>::data[static_cast<int>(b.dataType())]));
+ return fmt::format("Tensor({}, dims = {}, dtype = {})", b.toString(-1, 7), b.dims(), b.dataType());
})
.def("__len__", [](Tensor& b) -> size_t{
return b.size();
--
GitLab