Skip to content

Commit 3959519

Browse files
authored
TYP: Adjust typing for mypy failures (#292)
1 parent 7f82932 commit 3959519

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

src/fmu/tools/rms/copy_rms_param_to_ertbox_grid.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -763,7 +763,7 @@ def assign_undefined_constant(
763763
ertbox_values_3d_masked: np.ma.MaskedArray[Any, Any],
764764
value: float,
765765
debug_level: int = DEBUG_OFF,
766-
) -> np.ma.MaskedArray:
766+
) -> np.ndarray:
767767
if debug_level >= DEBUG_VERY_VERBOSE:
768768
print(f"--- All inactive values set to:{value}")
769769
return ertbox_values_3d_masked.filled(value)
@@ -807,7 +807,7 @@ def assign_undefined_vertical(
807807
nz_ertbox: int,
808808
ertbox_values_3d_masked: np.ma.MaskedArray[Any, Any],
809809
fill_value: float,
810-
) -> np.ma.MaskedArray[Any, Any]:
810+
) -> np.ndarray:
811811
k_indices = np.arange(nz_ertbox)
812812
for i in range(nx):
813813
for j in range(ny):
@@ -981,11 +981,11 @@ def extrapolate_values_for_zone(
981981

982982

983983
def add_noise_to_undefined_grid_cell_values(
984-
ertbox_values_3d: np.ma.MaskedArray[Any, Any],
984+
ertbox_values_3d: np.ndarray,
985985
undefined_grid_cells: np.ndarray,
986986
seed: int,
987987
max_relative_noise: float = 0.05,
988-
) -> np.ma.MaskedArray[Any, Any]:
988+
) -> np.ndarray:
989989
mean = np.mean(ertbox_values_3d)
990990
low = 0.0
991991
high = math.fabs(mean) * max_relative_noise

0 commit comments

Comments
 (0)