Skip to content

Commit 80976fa

Browse files
committed
fix typing issues
1 parent 44f8692 commit 80976fa

1 file changed

Lines changed: 10 additions & 10 deletions

File tree

src/pyuvdata/uvdata/uvdata.py

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -43,11 +43,11 @@
4343

4444
def flt_ind_str_arr(
4545
*,
46-
fltarr: npt.NDArray[float],
47-
intarr: npt.NDArray[int],
48-
flt_tols: tuple(float, float),
46+
fltarr: npt.NDArray[np.floating],
47+
intarr: npt.NDArray[np.int_],
48+
flt_tols: tuple[float, float],
4949
flt_first: bool = True,
50-
) -> npt.NDArray[str]:
50+
) -> npt.NDArray[np.str_]:
5151
"""
5252
Create a string array built from float and integer arrays for matching.
5353

@@ -82,8 +82,8 @@ def flt_ind_str_arr(
8282

8383

8484
def _add_freq_order(
85-
spw_id: npt.NDArray[int], freq_arr: npt.NDArray[float]
86-
) -> npt.NDArray[int]:
85+
spw_id: npt.NDArray[np.int_], freq_arr: npt.NDArray[np.floating]
86+
) -> npt.NDArray[np.int_]:
8787
"""
8888
Get the sorting order for the frequency axis after an add.
8989

@@ -125,8 +125,8 @@ def _axis_add_helper(
125125
this: UVData,
126126
other: UVData,
127127
axis_name: str,
128-
other_inds: npt.NDArray[int],
129-
final_order: npt.NDArray[int] | None = None,
128+
other_inds: npt.NDArray[np.int_],
129+
final_order: npt.NDArray[np.int_] | None = None,
130130
):
131131
"""
132132
Combine UVParameter objects with a single axis along an axis.
@@ -5569,7 +5569,7 @@ def fix_phase(self, *, use_ant_pos=True):
55695569
use_ant_pos=False,
55705570
)
55715571

5572-
def blt_str_arr(self) -> npt.NDArray[str]:
5572+
def blt_str_arr(self) -> npt.NDArray[np.str_]:
55735573
"""Create a string array with baseline and time info for matching purposes."""
55745574
return flt_ind_str_arr(
55755575
fltarr=self.time_array,
@@ -5578,7 +5578,7 @@ def blt_str_arr(self) -> npt.NDArray[str]:
55785578
flt_first=True,
55795579
)
55805580

5581-
def spw_freq_str_arr(self) -> npt.NDArray[str]:
5581+
def spw_freq_str_arr(self) -> npt.NDArray[np.str_]:
55825582
"""Create a string array with spw and freq info for matching purposes."""
55835583
return flt_ind_str_arr(
55845584
fltarr=self.freq_array,

0 commit comments

Comments
 (0)