From b7a7c5ae3da304e4c04bd5bfb7ff32c55f228bbb Mon Sep 17 00:00:00 2001 From: Mauko Quiroga Date: Sun, 5 Jan 2020 18:59:00 +0100 Subject: [PATCH] Utilise numpy.ljust au lieu de startswith --- openfisca_france/model/prestations/logement_social.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/openfisca_france/model/prestations/logement_social.py b/openfisca_france/model/prestations/logement_social.py index c77b9d8325..e6420be41c 100644 --- a/openfisca_france/model/prestations/logement_social.py +++ b/openfisca_france/model/prestations/logement_social.py @@ -1,5 +1,4 @@ -from numpy import isin, logical_not as not_, select -from numpy.core.defchararray import startswith +from numpy import char, isin, logical_not as not_, select from openfisca_core.indexed_enums import Enum from openfisca_core.periods import MONTH @@ -69,7 +68,7 @@ class zone_logement_social(Variable): def formula(menage, period): depcom = menage('depcom', period) in_paris_communes_limitrophes = isin(depcom, paris_communes_limitrophes) - in_idf = isin([True], startswith(depcom, departements_idf)) + in_idf = isin(char.ljust(depcom, 2), departements_idf) return select( [