Skip to content

Commit 0ac4575

Browse files
committed
fix: winsorize issue when only one percentile is given
1 parent 8007217 commit 0ac4575

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

eis_toolkit/transformations/winsorize.py

+6-4
Original file line numberDiff line numberDiff line change
@@ -150,12 +150,14 @@ def winsorize( # type: ignore[no-any-unimported]
150150
current_transform = f"transformation {i + 1}"
151151
current_settings = {
152152
"band_origin": bands[i],
153-
"percentile_lower": cast_scalar_to_int(percentiles[i][0]),
154-
"percentile_upper": cast_scalar_to_int(percentiles[i][1]),
155-
"calculated_lower": cast_scalar_to_int(calculated_lower),
156-
"calculated_upper": cast_scalar_to_int(calculated_upper),
157153
"nodata": cast_scalar_to_int(nodata),
158154
}
155+
if calculated_lower:
156+
current_settings["percentile_lower"] = cast_scalar_to_int(percentiles[i][0])
157+
current_settings["calculated_lower"] = cast_scalar_to_int(calculated_lower)
158+
if calculated_upper:
159+
current_settings["percentile_upper"] = cast_scalar_to_int(percentiles[i][1])
160+
current_settings["calculated_upper"] = cast_scalar_to_int(calculated_upper)
159161

160162
out_settings[current_transform] = current_settings
161163

0 commit comments

Comments
 (0)