From 21fb2d845a6c3f0709a20a758f20088d59e9be36 Mon Sep 17 00:00:00 2001 From: Altay Sansal Date: Mon, 7 Aug 2023 12:26:55 -0500 Subject: [PATCH] lint again --- src/mdio/converters/segy.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/src/mdio/converters/segy.py b/src/mdio/converters/segy.py index 752c7a4c..6eadc919 100644 --- a/src/mdio/converters/segy.py +++ b/src/mdio/converters/segy.py @@ -66,9 +66,12 @@ def grid_density_qc(grid: Grid, num_traces: int) -> None: Args: grid: The grid instance to check. num_traces: Expected number of traces. + + Raises: + GridTraceCountError: When the grid is too sparse. """ grid_traces = np.prod(grid.shape[:-1], dtype=np.uint64) # Exclude sample - dims = {name: shape for name, shape in zip(grid.dim_names, grid.shape)} + dims = {k: v for k, v in zip(grid.dim_names, grid.shape)} # noqa: B905 logger.debug(f"Dimensions: {dims}") logger.debug(f"num_traces = {num_traces}") @@ -84,7 +87,7 @@ def grid_density_qc(grid: Grid, num_traces: int) -> None: f"Grid shape: {grid.shape} but SEG-Y tracecount: {num_traces}. " "This grid is very sparse and most likely user error with indexing." ) - raise ValueError(msg) + raise GridTraceCountError(msg) # Warning if we have above 50% sparsity. if grid_traces > 2 * num_traces: