Skip to content

Commit 53a9454

Browse files
committed
Added back missing type-hinting
1 parent da71b6d commit 53a9454

File tree

2 files changed

+15
-13
lines changed

2 files changed

+15
-13
lines changed

src/mdio/segy/creation.py

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -43,15 +43,15 @@ def make_segy_factory(
4343
)
4444

4545

46-
def mdio_spec_to_segy(
47-
mdio_path_or_buffer,
48-
output_segy_path,
49-
access_pattern,
50-
output_endian,
51-
storage_options,
52-
new_chunks,
53-
backend,
54-
):
46+
def mdio_spec_to_segy( # noqa: PLR0913 DOC107
47+
mdio_path_or_buffer: str | Path,
48+
output_segy_path: str | Path,
49+
access_pattern: str,
50+
output_endian: Endianness,
51+
storage_options: dict[str, Any],
52+
new_chunks: tuple[int, ...],
53+
backend: str,
54+
) -> tuple[MDIOReader, SegyFactory]:
5555
"""Create SEG-Y file without any traces given MDIO specification.
5656
5757
This function opens an MDIO file, gets some relevant information for SEG-Y files,

src/mdio/segy/geometry.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -209,7 +209,7 @@ def create_counter(
209209
total_depth: int,
210210
unique_headers: dict[str, NDArray],
211211
header_names: list[str],
212-
):
212+
) -> int | dict:
213213
"""Helper function to create dictionary tree for counting trace key for auto index."""
214214
if depth == total_depth:
215215
return 0
@@ -230,8 +230,8 @@ def create_trace_index(
230230
counter: dict,
231231
index_headers: HeaderArray,
232232
header_names: list,
233-
dtype=np.int16,
234-
):
233+
dtype: np.dtype = np.int16,
234+
) -> None | HeaderArray:
235235
"""Update dictionary counter tree for counting trace key for auto index."""
236236
if depth == 0:
237237
# If there's no hierarchical depth, no tracing needed.
@@ -262,7 +262,9 @@ def create_trace_index(
262262
return index_headers
263263

264264

265-
def analyze_non_indexed_headers(index_headers: HeaderArray, dtype=np.int16) -> NDArray:
265+
def analyze_non_indexed_headers(
266+
index_headers: HeaderArray, dtype: np.dtype = np.int16
267+
) -> NDArray:
266268
"""Check input headers for SEG-Y input to help determine geometry.
267269
268270
This function reads in trace_qc_count headers and finds the unique cable values.

0 commit comments

Comments
 (0)