Cleanup

Author: BrianMichell

src/mdio/converters/segy.py

@@ -142,7 +142,6 @@ def _scan_for_headers(
     segy_file_info: SegyFileInfo,
     template: AbstractDatasetTemplate,
     grid_overrides: dict[str, Any] | None = None,
-    calculate_checksum: bool = False,
 ) -> tuple[list[Dimension], SegyHeaderArray, int | None]:
     """Extract trace dimensions and index headers from the SEG-Y file.
 
@@ -158,10 +157,9 @@ def _scan_for_headers(
         template=template,
         chunksize=full_chunk_size,
         grid_overrides=grid_overrides,
-        calculate_checksum=calculate_checksum,
     )
 
-    if calculate_checksum:
+    if should_calculate_checksum():
         segy_dimensions, chunk_size, segy_headers, trace_data_crc32c = grid_plan_result
     else:
         segy_dimensions, chunk_size, segy_headers = grid_plan_result
@@ -529,16 +527,11 @@ def segy_to_mdio(  # noqa PLR0913
     }
     segy_file_info = get_segy_file_info(segy_file_kwargs)
 
-    # Check if checksum calculation should be performed
-    # This checks both the env var and library availability
-    calculate_checksum = should_calculate_checksum()
-
     segy_dimensions, segy_headers, trace_data_crc32c = _scan_for_headers(
         segy_file_kwargs,
         segy_file_info,
         template=mdio_template,
         grid_overrides=grid_overrides,
-        calculate_checksum=calculate_checksum,
     )
     grid = _build_and_check_grid(segy_dimensions, segy_file_info, segy_headers)
 

src/mdio/segy/utilities.py

@@ -34,7 +34,6 @@ def get_grid_plan(  # noqa:  C901, PLR0913
     template: AbstractDatasetTemplate,
     return_headers: bool = False,
     grid_overrides: dict[str, Any] | None = None,
-    calculate_checksum: bool = False,
 ) -> (
     tuple[list[Dimension], tuple[int, ...]]
     | tuple[list[Dimension], tuple[int, ...], HeaderArray]
@@ -60,8 +59,10 @@ def get_grid_plan(  # noqa:  C901, PLR0913
     Returns:
         All index dimensions and chunksize, optionally with header values and/or checksum.
     """
+    calculate_checksum = False
     if should_calculate_checksum() and is_checksum_available():
         from mdio.segy.checksum import parse_headers
+        calculate_checksum = True
     else:
         from mdio.segy.parsers import parse_headers