Quick bug fix for 2d numpy arrays of strings

MAVENSDC · May 10, 2024 · bf90608 · bf90608
1 parent 6ec09a7
commit bf90608
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cdflib/cdfwrite.py b/cdflib/cdfwrite.py
@@ -2336,7 +2336,7 @@ def _convert_data(self, data_type: int, num_elems: int, num_values: int, indata:
             return recs, struct.pack(form2, *datau)
         elif isinstance(indata, np.ndarray):
             if data_type == self.CDF_CHAR or data_type == self.CDF_UCHAR:
-                size = indata.size
+                size = len(np.atleast_1d(indata))
                 odata = ""
                 if size >= 1:
                     for x in range(0, size):