fixing unit tests

SocialFinanceDigitalLabs · May 24, 2024 · ae9fce4 · ae9fce4
1 parent c47068c
commit ae9fce4
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/dm_regional_app/builder.py b/dm_regional_app/builder.py
@@ -44,8 +44,9 @@ def scenario(
  name: Optional[str] = None,
  description: Optional[str] = None,
  historic_filters: Optional[str] = None,
- prediction_filters: Optional[str] = None,
  prediction_parameters: Optional[str] = None,
+ adjusted_rates: Optional[str] = None,
+ adjusted_numbers: Optional[str] = None,
  historic_stock: Optional[str] = None,
  adjusted_costs: Optional[str] = None,
  **kwargs,
@@ -55,8 +56,9 @@ def scenario(
  name=name or self.fake.name(),
  description=description or self.fake.text(),
  historic_filters=historic_filters or self.fake.text(),
- prediction_filters=prediction_filters or self.fake.text(),
  prediction_parameters=prediction_parameters or self.fake.text(),
+ adjusted_rates=adjusted_rates or self.fake.text(),
+ adjusted_numbers=adjusted_numbers or self.fake.text(),
  historic_stock=historic_stock or self.fake.text(),
  adjusted_costs=adjusted_costs or self.fake.text(),
  **kwargs,