Skip to content

Commit dc4de48

Browse files
author
Mauko Quiroga
committed
Add weeks to holders
1 parent c557909 commit dc4de48

1 file changed

Lines changed: 5 additions & 1 deletion

File tree

openfisca_core/holders.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212
from openfisca_core.data_storage import InMemoryStorage, OnDiskStorage
1313
from openfisca_core.errors import PeriodMismatchError
1414
from openfisca_core.indexed_enums import Enum
15-
from openfisca_core.periods import MONTH, YEAR, ETERNITY
15+
from openfisca_core.periods import WEEK, MONTH, YEAR, ETERNITY
1616
from openfisca_core.tools import eval_expression
1717

1818
log = logging.getLogger(__name__)
@@ -209,6 +209,10 @@ def _set(self, period, value):
209209
if self.variable.definition_period != ETERNITY:
210210
if period is None:
211211
raise ValueError('A period must be specified to set values, except for variables with ETERNITY as as period_definition.')
212+
213+
if self.variable.definition_period == WEEK:
214+
period = periods.period(f"{WEEK}:{period}")
215+
212216
if (self.variable.definition_period != period.unit or period.size > 1):
213217
name = self.variable.name
214218
period_size_adj = f'{period.unit}' if (period.size == 1) else f'{period.size}-{period.unit}s'

0 commit comments

Comments
 (0)