minor Eigen fix for debug mode compilation

wjakob · May 3, 2023 · a965f5b · a965f5b
1 parent 74235fa
commit a965f5b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/include/nanobind/eigen/dense.h b/include/nanobind/eigen/dense.h
@@ -226,7 +226,8 @@ struct type_caster<Eigen::Map<T, Options, StrideType>, enable_if_t<is_eigen_plai
 
     operator Map() {
         NDArray &t = caster.value;
-        return Map(t.data(), t.shape(0), t.shape(1), strides());
+        return Map(t.data(), t.shape(0), t.ndim() == 1 ? 1 : t.shape(1),
+                   strides());
     }
 };