Skip to content

Commit cc80cba

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 91b4b9c commit cc80cba

File tree

1 file changed

+9
-15
lines changed

1 file changed

+9
-15
lines changed

src/_gettsim/groupings.py

Lines changed: 9 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@ def create_groupings() -> dict[str, Callable]:
1010
"eg_id": eg_id_numpy,
1111
"fg_id": fg_id_numpy,
1212
"sn_id": sn_id_numpy,
13-
"ehe_id": ehe_id_numpy
13+
"ehe_id": ehe_id_numpy,
1414
}
1515

1616

@@ -40,8 +40,8 @@ def bg_id_numpy(
4040

4141

4242
def eg_id_numpy(
43-
p_id: numpy.ndarray,
44-
p_id_einstandspartner: numpy.ndarray,
43+
p_id: numpy.ndarray,
44+
p_id_einstandspartner: numpy.ndarray,
4545
) -> numpy.ndarray:
4646
"""
4747
Compute the ID of the Einstandsgemeinschaft for each person.
@@ -54,8 +54,8 @@ def eg_id_numpy(
5454
current_p_id_einstandspartner = p_id_einstandspartner[index]
5555

5656
if (
57-
current_p_id_einstandspartner >= 0
58-
and current_p_id_einstandspartner in p_id_to_eg_id
57+
current_p_id_einstandspartner >= 0
58+
and current_p_id_einstandspartner in p_id_to_eg_id
5959
):
6060
result.append(p_id_to_eg_id[current_p_id_einstandspartner])
6161
continue
@@ -69,8 +69,8 @@ def eg_id_numpy(
6969

7070

7171
def ehe_id_numpy(
72-
p_id: numpy.ndarray,
73-
p_id_ehepartner: numpy.ndarray,
72+
p_id: numpy.ndarray,
73+
p_id_ehepartner: numpy.ndarray,
7474
):
7575
"""
7676
Compute the ID of the Ehe for each person.
@@ -82,10 +82,7 @@ def ehe_id_numpy(
8282
for index, current_p_id in enumerate(p_id):
8383
current_p_id_ehepartner = p_id_ehepartner[index]
8484

85-
if (
86-
current_p_id_ehepartner >= 0
87-
and current_p_id_ehepartner in p_id_to_ehe_id
88-
):
85+
if current_p_id_ehepartner >= 0 and current_p_id_ehepartner in p_id_to_ehe_id:
8986
result.append(p_id_to_ehe_id[current_p_id_ehepartner])
9087
continue
9188

@@ -190,10 +187,7 @@ def sn_id_numpy(
190187
current_p_id_ehepartner = p_id_ehepartner[index]
191188
current_gemeinsam_veranlagt = gemeinsam_veranlagt[index]
192189

193-
if (
194-
current_p_id_ehepartner >= 0
195-
and current_p_id_ehepartner in p_id_to_sn_id
196-
):
190+
if current_p_id_ehepartner >= 0 and current_p_id_ehepartner in p_id_to_sn_id:
197191
gemeinsam_veranlagt_ehepartner = p_id_to_gemeinsam_veranlagt[
198192
current_p_id_ehepartner
199193
]

0 commit comments

Comments
 (0)