Skip to content

Commit d6cbbd7

Browse files
pre-commit-ci[bot]aaronspring
authored andcommitted
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 128dd50 commit d6cbbd7

11 files changed

+46
-35
lines changed

climpred/bias_removal.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -338,9 +338,9 @@ def gaussian_bias_removal(
338338
hindcast_train._datasets["initialized"] = hindcast.get_initialized().sel(
339339
init=train_time_init
340340
)
341-
hindcast_test._datasets[
342-
"initialized"
343-
] = hindcast.get_initialized().drop_sel(init=train_time_init)
341+
hindcast_test._datasets["initialized"] = (
342+
hindcast.get_initialized().drop_sel(init=train_time_init)
343+
)
344344
else:
345345
assert train_test_split in ["unfair", "unfair-cv"]
346346
hindcast_train = hindcast

climpred/bootstrap.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -484,9 +484,9 @@ def resample_skill_loop(self, iterations, resample_dim, verify_kwargs):
484484
# resample uninitialized
485485
if not self.get_uninitialized():
486486
# warn not found and therefore generate
487-
self_for_loop._datasets[
488-
"uninitialized"
489-
] = self.generate_uninitialized().get_uninitialized()
487+
self_for_loop._datasets["uninitialized"] = (
488+
self.generate_uninitialized().get_uninitialized()
489+
)
490490
else:
491491
self_for_loop._datasets["uninitialized"] = _resample(
492492
self.get_uninitialized(), "member"

climpred/classes.py

Lines changed: 20 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -1456,9 +1456,11 @@ def verify(
14561456
reference = _check_valid_reference(reference)
14571457
input_dict = {
14581458
"ensemble": self._datasets["initialized"],
1459-
"control": self._datasets["control"]
1460-
if isinstance(self._datasets["control"], xr.Dataset)
1461-
else None,
1459+
"control": (
1460+
self._datasets["control"]
1461+
if isinstance(self._datasets["control"], xr.Dataset)
1462+
else None
1463+
),
14621464
"init": True,
14631465
}
14641466
result = self._apply_climpred_function(
@@ -1535,9 +1537,11 @@ def _compute_uninitialized(
15351537
)
15361538
input_dict = {
15371539
"ensemble": self._datasets["uninitialized"],
1538-
"control": self._datasets["control"]
1539-
if isinstance(self._datasets["control"], xr.Dataset)
1540-
else None,
1540+
"control": (
1541+
self._datasets["control"]
1542+
if isinstance(self._datasets["control"], xr.Dataset)
1543+
else None
1544+
),
15411545
"init": False,
15421546
}
15431547
if dim is None:
@@ -1664,9 +1668,11 @@ def _compute_climatology(
16641668
"""
16651669
input_dict = {
16661670
"ensemble": self._datasets["initialized"],
1667-
"control": self._datasets["control"]
1668-
if isinstance(self._datasets["control"], xr.Dataset)
1669-
else None,
1671+
"control": (
1672+
self._datasets["control"]
1673+
if isinstance(self._datasets["control"], xr.Dataset)
1674+
else None
1675+
),
16701676
"init": True,
16711677
}
16721678
if dim is None:
@@ -2877,11 +2883,9 @@ def remove_bias(
28772883
if len(self.get_initialized().coords["valid_time"].dims) == 1 and set(
28782884
["init", "lead"]
28792885
).issubset(set(self.get_initialized().dims)):
2880-
self._datasets["initialized"].coords[
2881-
"valid_time"
2882-
] = add_time_from_init_lead(
2883-
self.get_initialized().drop("valid_time")
2884-
).coords[
2885-
"valid_time"
2886-
]
2886+
self._datasets["initialized"].coords["valid_time"] = (
2887+
add_time_from_init_lead(
2888+
self.get_initialized().drop("valid_time")
2889+
).coords["valid_time"]
2890+
)
28872891
return self

climpred/graphics.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -415,9 +415,11 @@ def _verif_dates_xr(hindcast, alignment, reference, date2num_units):
415415
hindcast.get_observations(),
416416
alignment,
417417
reference=reference,
418-
hist=hindcast.get_uninitialized()
419-
if isinstance(hindcast.get_uninitialized(), xr.Dataset)
420-
else None,
418+
hist=(
419+
hindcast.get_uninitialized()
420+
if isinstance(hindcast.get_uninitialized(), xr.Dataset)
421+
else None
422+
),
421423
)
422424

423425
verif_dates_xr = xr.concat(

climpred/tests/test_bias_removal.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
"""Test bias_removal.py."""
2+
23
import copy
34

45
import numpy as np

climpred/tests/test_graphics.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
"""Test graphics.py and PredictionEnsemble.plot()."""
2+
23
import pytest
34

45
from climpred import HindcastEnsemble, PerfectModelEnsemble

climpred/tests/test_metrics_perfect.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -167,9 +167,11 @@ def f(x):
167167
skill = he.verify(
168168
metric=metric,
169169
comparison="m2o",
170-
dim=["member", "init"]
171-
if metric in probabilistic_metrics_requiring_more_than_member_dim
172-
else "member",
170+
dim=(
171+
["member", "init"]
172+
if metric in probabilistic_metrics_requiring_more_than_member_dim
173+
else "member"
174+
),
173175
alignment=alignment,
174176
**metric_kwargs,
175177
).SST

climpred/tests/test_perfect_model_prediction.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,5 @@
11
"""Test compute_perfect_model."""
22

3-
43
import pytest
54
import xarray as xr
65

climpred/tests/test_relative_entropy.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
"""Test relative_entropy.py"""
2+
23
import pytest
34

45
from climpred.graphics import plot_relative_entropy

climpred/tests/test_uninitialized.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
"""Test compute_uninitialized."""
2+
23
import pytest
34

45
from climpred.constants import VALID_ALIGNMENTS

climpred/utils.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -86,9 +86,9 @@ def assign_attrs(
8686
skill[v].attrs.update(ds[v].attrs)
8787

8888
# climpred info
89-
skill.attrs[
90-
"prediction_skill_software"
91-
] = "climpred https://climpred.readthedocs.io/"
89+
skill.attrs["prediction_skill_software"] = (
90+
"climpred https://climpred.readthedocs.io/"
91+
)
9292
if function_name:
9393
skill.attrs["skill_calculated_by_function"] = function_name
9494
if "init" in ds.coords and "init" not in skill.dims:
@@ -138,9 +138,9 @@ def assign_attrs(
138138
if "pers_sig" in kwargs:
139139
if kwargs["pers_sig"] is not None:
140140
_, ci_low_pers, ci_high_pers = _p_ci_from_sig(kwargs["pers_sig"])
141-
kwargs[
142-
"confidence_interval_levels_persistence"
143-
] = f"{ci_high_pers}-{ci_low_pers}"
141+
kwargs["confidence_interval_levels_persistence"] = (
142+
f"{ci_high_pers}-{ci_low_pers}"
143+
)
144144
# check for none attrs and remove
145145
del_list = []
146146
for key, value in kwargs.items():

0 commit comments

Comments
 (0)