refactor: Update eigen_shape() function to handle vectors in broadcas…

…t.hpp
CNES · Jul 3, 2024 · 278bf72 · 278bf72
1 parent a716ce7
commit 278bf72
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/include/fes/detail/broadcast.hpp b/include/fes/detail/broadcast.hpp
@@ -9,6 +9,8 @@
 #include <stdexcept>
 #include <string>
 
+#include "fes/eigen.hpp"
+
 namespace fes {
 namespace detail {
 
@@ -18,7 +20,11 @@ namespace detail {
 template <typename Array>
 auto eigen_shape(const Array& array) -> std::string {
   std::stringstream ss;
-  ss << "(" << array.rows() << ", " << array.cols() << ")";
+  if (is_vector<Array>()) {
+    ss << "(" << array.size() << ")";
+  } else {
+    ss << "(" << array.rows() << ", " << array.cols() << ")";
+  }
   return ss.str();
 }