Skip to content

Commit

Permalink
preprocessor and annotate_households settings
Browse files Browse the repository at this point in the history
  • Loading branch information
dhensle committed Mar 8, 2024
1 parent c0357fb commit 3c7bba4
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions activitysim/abm/models/auto_ownership.py
Original file line number Diff line number Diff line change
Expand Up @@ -26,8 +26,8 @@ class AutoOwnershipSettings(LogitComponentSettings):
"""
Settings for the `auto_ownership` component.
"""

# This model is relatively simple and has no unique settings
preprocessor: PreprocessorSettings | None = None
annotate_households: PreprocessorSettings | None = None


@workflow.step
Expand Down Expand Up @@ -66,7 +66,7 @@ def auto_ownership_simulate(
logger.info("Running %s with %d households", trace_label, len(choosers))

# - preprocessor
preprocessor_settings = model_settings.get("preprocessor", None)
preprocessor_settings = model_settings.preprocessor
if preprocessor_settings:

locals_d = {}
Expand Down Expand Up @@ -115,7 +115,7 @@ def auto_ownership_simulate(
"auto_ownership", households.auto_ownership, value_counts=True
)

if model_settings.get("annotate_households"):
if model_settings.annotate_households:
annotate.annotate_households(model_settings, trace_label)

if trace_hh_id:
Expand Down

0 comments on commit 3c7bba4

Please sign in to comment.