Utilise numpy.ljust au lieu de startswith

openfisca · Jan 6, 2020 · b7a7c5a · b7a7c5a
1 parent 611e762
commit b7a7c5a
Showing 1 changed file with 2 additions and 3 deletions.
diff --git 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(
             [