Fix pre-commit

Author: dmitriyrepin

src/mdio/api/create.py

@@ -50,6 +50,9 @@ def create_empty(  # noqa PLR0913
         headers: SEG-Y v1.0 trace headers. Defaults to None.
         overwrite: Whether to overwrite the output file if it already exists. Defaults to False.
 
+    Returns:
+        The output MDIO dataset.
+
     Raises:
         FileExistsError: If the output location already exists and overwrite is False.
     """
@@ -86,7 +89,7 @@ def create_empty(  # noqa PLR0913
 
     # Write the dimension coordinates and trace mask
     xr_dataset = xr_dataset[drop_vars_delayed + ["trace_mask"]]
-    
+
     if output_path is not None:
         to_mdio(xr_dataset, output_path=output_path, mode="r+", compute=True)
 

tests/integration/test_z_create_empty.py

@@ -91,7 +91,7 @@ class TestCreateEmptyMdio:
     """Tests for create_empty_mdio function."""
 
     @classmethod
-    def _create_empty_mdio(cls, create_headers: bool, output_path: Path, overwrite: bool = True) -> None:
+    def _create_empty_mdio(cls, create_headers: bool, output_path: Path, overwrite: bool = True) -> xr_Dataset:
         """Create a temporary empty MDIO file for testing."""
         # Create the grid with the specified dimensions
         dims = [
@@ -104,14 +104,13 @@ def _create_empty_mdio(cls, create_headers: bool, output_path: Path, overwrite:
         # The HeaderSpec can be either standard or customized.
         headers = get_teapot_segy_spec().trace.header if create_headers else None
         # Create an empty MDIO v1 metric post-stack 3D time velocity dataset
-        xr_dataset = create_empty(
+        return create_empty(
             mdio_template=PostStack3DVelocityTemplate(data_domain="time", is_metric=True),
             dimensions=dims,
             output_path=output_path,
             headers=headers,
             overwrite=overwrite,
         )
-        return xr_dataset
 
     @classmethod
     def validate_teapod_dataset_metadata(cls, ds: xr_Dataset, is_velocity: bool) -> None: