@@ -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
4242def 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
7171def 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