Run pre-commit

Author: dmitriyrepin

src/mdio/creators/mdio.py

@@ -9,22 +9,13 @@
 from mdio.api.io import _normalize_path
 from mdio.api.io import open_mdio
 from mdio.api.io import to_mdio
-from mdio.builder.template_registry import TemplateRegistry
-from mdio.builder.xarray_builder import to_xarray_dataset
-from mdio.converters.segy import populate_dim_coordinates
-from mdio.converters.type_converter import to_structured_type
-from mdio.core.grid import Grid
 
 if TYPE_CHECKING:
     from pathlib import Path
 
-    from segy.schema import HeaderSpec
     from upath import UPath
     from xarray import Dataset as xr_Dataset
 
-    from mdio.builder.schemas import Dataset
-    from mdio.core.dimension import Dimension
-
 
 def create_empty_like(  # noqa PLR0913
     input_path: UPath | Path | str,

tests/conftest.py

@@ -46,6 +46,7 @@ def zarr_tmp(tmp_path_factory: pytest.TempPathFactory) -> Path:
     """Make a temp file for the output MDIO."""
     return tmp_path_factory.mktemp(r"mdio")
 
+
 @pytest.fixture(scope="session")
 def teapot_mdio_tmp(tmp_path_factory: pytest.TempPathFactory) -> Path:
     """Make a temp file for the output MDIO."""
@@ -74,8 +75,7 @@ def segy_export_tmp(tmp_path_factory: pytest.TempPathFactory) -> Path:
 @pytest.fixture(scope="class")
 def empty_mdio_with_headers(tmp_path_factory: pytest.TempPathFactory) -> Path:
     """Make a temp file for empty MDIO testing."""
-    path = tmp_path_factory.mktemp(r"empty_with_headers.mdio")
-    return path
+    return tmp_path_factory.mktemp(r"empty_with_headers.mdio")
 
 
 # @pytest.fixture(scope="session")

tests/integration/test_create_empty.py

@@ -2,7 +2,6 @@
 
 from __future__ import annotations
 
-import math
 from typing import TYPE_CHECKING
 
 import numpy as np
@@ -12,13 +11,6 @@
 from segy.schema import ScalarType
 from segy.standards import get_segy_standard
 
-from mdio.builder.schemas.v1.units import LengthUnitEnum
-from mdio.builder.schemas.v1.units import LengthUnitModel
-from mdio.builder.schemas.v1.units import SpeedUnitEnum
-from mdio.builder.schemas.v1.units import SpeedUnitModel
-from mdio.builder.schemas.v1.units import TimeUnitEnum
-from mdio.builder.schemas.v1.units import TimeUnitModel
-
 if TYPE_CHECKING:
     from pathlib import Path
 
@@ -30,12 +22,6 @@
 from tests.integration.testing_helpers import validate_variable
 
 from mdio import __version__
-from mdio.api.io import open_mdio
-from mdio.api.io import to_mdio
-from mdio.builder.schemas.v1.stats import CenteredBinHistogram
-from mdio.builder.schemas.v1.stats import SummaryStatistics
-from mdio.converters.mdio import mdio_to_segy
-from mdio.core import Dimension
 from mdio.creators.mdio import create_empty_like
 
 

tests/integration/test_segy_roundtrip_teapot.py

@@ -42,8 +42,10 @@ def set_env_vars(monkeypatch: Generator[pytest.MonkeyPatch]) -> None:
 
 @pytest.fixture
 def teapot_segy_spec() -> SegySpec:
+    """Return the customized SEG-Y specification for the teapot dome dataset."""
     return get_teapot_segy_spec()
 
+
 def get_teapot_segy_spec() -> SegySpec:
     """Return the customized SEG-Y specification for the teapot dome dataset."""
     teapot_fields = [