Skip to content

Commit

Permalink
read coefficients of mandatory tour frequency
Browse files Browse the repository at this point in the history
  • Loading branch information
vivekyadav26 committed Nov 9, 2023
1 parent 2901363 commit 2772d7e
Showing 1 changed file with 10 additions and 2 deletions.
12 changes: 10 additions & 2 deletions activitysim/core/simulate.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,6 +14,9 @@
import numpy as np
import pandas as pd

from activitysim.abm.models.mandatory_tour_frequency import (
MandatoryTourFrequencySettings,
)
from activitysim.core import (
assign,
chunk,
Expand Down Expand Up @@ -147,7 +150,10 @@ def read_model_spec(filesystem: configuration.FileSystem, file_name: Path | str)

def read_model_coefficients(
filesystem: configuration.FileSystem,
model_settings: LogitComponentSettings | dict[str, Any] | None = None,
model_settings: LogitComponentSettings
| MandatoryTourFrequencySettings
| dict[str, Any]
| None = None,
file_name: Path | str | None = None,
) -> pd.DataFrame:
"""
Expand All @@ -159,7 +165,9 @@ def read_model_coefficients(
assert file_name is not None
else:
assert file_name is None
if isinstance(model_settings, LogitComponentSettings):
if isinstance(model_settings, LogitComponentSettings) | isinstance(
model_settings, MandatoryTourFrequencySettings
):
file_name = model_settings.COEFFICIENTS
else:
assert (
Expand Down

0 comments on commit 2772d7e

Please sign in to comment.