Ensure all encodings are captured

Author: BrianMichell

src/mdio/converters/segy.py

@@ -229,6 +229,8 @@ def populate_non_dim_coordinates(
     """Populate the xarray dataset with coordinate variables."""
     not_null = grid.map[:] != UINT32_MAX
     for c_name, c_values in coordinates.items():
+        # Since we haven't serialized the dataset yet, if we don't explicitly capture the encoding
+        # we will lose it.
         encodings = dataset[c_name].encoding
         tmp_coords = np.full(
             not_null.shape, dtype=dataset[c_name].dtype, fill_value=_get_fill_value(dataset[c_name].dtype)
@@ -239,7 +241,7 @@ def populate_non_dim_coordinates(
             dataset[c_name].dims,
             tmp_coords,
             attrs=dataset[c_name].attrs,
-            encoding=encodings,  # Ensure we preserve all of the encodings.
+            encoding=encodings,
         )
         drop_vars_delayed.append(c_name)
     return dataset, drop_vars_delayed

src/mdio/segy/_workers.py

@@ -134,6 +134,7 @@ def trace_worker(  # noqa: PLR0913
             ds_to_write[hdr_key].dims,
             tmp_headers,
             attrs=ds_to_write[hdr_key].attrs,
+            encoding=ds_to_write[hdr_key].encoding,  # Not strictly necessary, but safer than not doing it.
         )
 
     # Get the sample dimension size from the data variable itself
@@ -153,6 +154,7 @@ def trace_worker(  # noqa: PLR0913
         ds_to_write[data_variable_name].dims,
         tmp_samples,
         attrs=ds_to_write[data_variable_name].attrs,
+        encoding=ds_to_write[data_variable_name].encoding,  # Not strictly necessary, but safer than not doing it.
     )
 
     out_path = output_location.uri