fix(data storage): added numpy type check for consistent integer and …

…float sizes (32-bit vs 64-bit) (modflowpy#2062) * fix(data storage): added numpy type check for consistent integer and float sizes * fix(numpy type check)
wpbonelli · Jan 17, 2024 · 97da396 · 97da396
1 parent 6899553
commit 97da396
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/flopy/mf6/data/mfdatastorage.py b/flopy/mf6/data/mfdatastorage.py
@@ -1407,6 +1407,16 @@ def store_internal(
                         message,
                         self._simulation_data.debug,
                     )
+                data_type = self.data_dimensions.structure.get_datum_type(True)
+                dt = self.layer_storage[layer].internal_data.dtype
+                if dt != data_type:
+                    self.layer_storage[
+                        layer
+                    ].internal_data = self.layer_storage[
+                        layer
+                    ].internal_data.astype(
+                        data_type
+                    )
             if not preserve_record:
                 self.layer_storage[layer].factor = multiplier
                 self.layer_storage[layer].iprn = print_format