Refactor: simplify and condense dictionary and function calls (#548)

Author: tasansal

src/mdio/commands/info.py

@@ -45,11 +45,7 @@ def info(mdio_path: str, output_format: str, access_pattern: str) -> None:
     """
     from mdio import MDIOReader
 
-    reader = MDIOReader(
-        mdio_path,
-        access_pattern=access_pattern,
-        return_metadata=True,
-    )
+    reader = MDIOReader(mdio_path, access_pattern=access_pattern, return_metadata=True)
 
     grid_dict = parse_grid(reader.grid)
     stats_dict = cast_stats(reader.stats)

src/mdio/converters/numpy.py

@@ -147,11 +147,7 @@ def numpy_to_mdio(  # noqa: PLR0913
     )
 
     dims = [Dimension(name=name, coords=index_coords[name]) for name in index_names]
-    create_conf = MDIOCreateConfig(
-        path=mdio_path_or_buffer,
-        grid=Grid(dims),
-        variables=[mdio_var],
-    )
+    create_conf = MDIOCreateConfig(path=mdio_path_or_buffer, grid=Grid(dims), variables=[mdio_var])
     create_empty(create_conf, overwrite, storage_options)
 
     writer = MDIOWriter(mdio_path_or_buffer, suffix, storage_options)

src/mdio/core/factory.py

@@ -171,13 +171,7 @@ def create_empty(
             chunk_key_encoding={"name": "v2", "separator": "/"},
         )
 
-    stats = {
-        "mean": 0,
-        "std": 0,
-        "rms": 0,
-        "min": 0,
-        "max": 0,
-    }
+    stats = {"mean": 0, "std": 0, "rms": 0, "min": 0, "max": 0}
 
     for key, value in stats.items():
         write_attribute(name=key, zarr_group=root_group, attribute=value)
@@ -238,11 +232,7 @@ def create_empty_like(
 
     config = MDIOCreateConfig(path=dest_path, grid=grid, variables=variables)
 
-    create_empty(
-        config=config,
-        overwrite=overwrite,
-        storage_options=storage_options_output,
-    )
+    create_empty(config=config, overwrite=overwrite, storage_options=storage_options_output)
 
     writer = MDIOWriter(dest_path, storage_options=storage_options_output)
     writer.text_header = src_meta_grp.attrs["text_header"]

src/mdio/exceptions.py

@@ -44,7 +44,6 @@ class WrongTypeError(MDIOError):
     def __init__(self, message: str, name: str = None, expected: str = None):
         if name is not None and expected is not None:
             extras = f"Got: {name} Expected: {expected}"
-
             message = f"{message} - {extras}"
 
         super().__init__(message)

src/mdio/segy/_workers.py

@@ -17,10 +17,7 @@
     from mdio.core import Grid
 
 
-def header_scan_worker(
-    segy_file: SegyFile,
-    trace_range: tuple[int, int],
-) -> HeaderArray:
+def header_scan_worker(segy_file: SegyFile, trace_range: tuple[int, int]) -> HeaderArray:
     """Header scan worker.
 
     If SegyFile is not open, it can either accept a path string or a handle that was opened in
@@ -90,7 +87,6 @@ def trace_worker(
     seq_trace_indices = grid.map[chunk_indices[:-1]]
 
     tmp_data = np.zeros(seq_trace_indices.shape + (grid.shape[-1],), dtype=data_array.dtype)
-
     tmp_metadata = np.zeros(seq_trace_indices.shape, dtype=metadata_array.dtype)
 
     del grid  # To save some memory

src/mdio/segy/blocked_io.py

@@ -32,12 +32,7 @@
 default_cpus = cpu_count(logical=True)
 
 
-def to_zarr(
-    segy_file: SegyFile,
-    grid: Grid,
-    data_array: Array,
-    header_array: Array,
-) -> dict:
+def to_zarr(segy_file: SegyFile, grid: Grid, data_array: Array, header_array: Array) -> dict:
     """Blocked I/O from SEG-Y to chunked `zarr.core.Array`.
 
     Args:
@@ -112,13 +107,7 @@ def to_zarr(
     glob_min = glob_min.min().astype("float64")
     glob_max = glob_max.max().astype("float64")
 
-    return {
-        "mean": glob_mean,
-        "std": glob_std,
-        "rms": glob_rms,
-        "min": glob_min,
-        "max": glob_max,
-    }
+    return {"mean": glob_mean, "std": glob_std, "rms": glob_rms, "min": glob_min, "max": glob_max}
 
 
 def segy_record_concat(

src/mdio/segy/creation.py

@@ -26,10 +26,7 @@
 logger = logging.getLogger(__name__)
 
 
-def make_segy_factory(
-    mdio: MDIOReader,
-    spec: SegySpec,
-) -> SegyFactory:
+def make_segy_factory(mdio: MDIOReader, spec: SegySpec) -> SegyFactory:
     """Generate SEG-Y factory from MDIO metadata."""
     grid = mdio.grid
     sample_dim = grid.select_dim("sample")