Update openfisca_france_data/erfs_fpr/input_data_builder/step_05_foye…

…r.py Co-authored-by: Mahdi Ben Jelloul <[email protected]>
openfisca · Nov 19, 2024 · d2937e4 · d2937e4
1 parent 01dc48b
commit d2937e4
Showing 1 changed file with 9 additions and 6 deletions.
diff --git a/openfisca_france_data/erfs_fpr/input_data_builder/step_05_foyer.py b/openfisca_france_data/erfs_fpr/input_data_builder/step_05_foyer.py
@@ -33,12 +33,15 @@ def build_foyer(temporary_store = None, year = None):
         how = "inner"
         )
 
-    decls['statut_decl1'] = np.where((decls.statut_decl1 != decls.statut_decl2),
-                                    np.where(decls.statut_decl1.isin([1,2]) | decls.statut_decl2.isin([1,2]),
-                                             1,
-                                             8
-                                             ),
-                                    decls.statut_decl1)
+    decls['statut_decl1'] = np.where(
+        (decls.statut_decl1 != decls.statut_decl2),
+        np.where(
+            decls.statut_decl1.isin([1,2]) | decls.statut_decl2.isin([1,2]),
+            1,
+            8
+            ),
+        decls.statut_decl1
+        )
 
     assert len(couples[couples.quifam.isin([0])]) == len(decls)
     couples = pd.merge(couples,decls[['idfam',"statut_decl1"]],on = 'idfam', how = 'inner')