Skip to content

Commit

Permalink
annotate_land_use_taz
Browse files Browse the repository at this point in the history
  • Loading branch information
jpn-- committed May 1, 2024
1 parent f0d165a commit 106e631
Show file tree
Hide file tree
Showing 5 changed files with 43 additions and 0 deletions.
2 changes: 2 additions & 0 deletions configs/resident/annotate_landuse_taz.csv
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
Description,Target,Expression
micro access time at taz,taz_microAccTime,"land_use.sort_values(by='MicroAccessTime',ascending=False).drop_duplicates('TAZ',keep='first').set_index('TAZ')['MicroAccessTime'].reindex(df.index)"
7 changes: 7 additions & 0 deletions configs/resident/initialize_landuse_taz.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,7 @@
annotate_tables:
- tablename: land_use_taz
annotate:
SPEC: annotate_landuse_taz
DF: land_use_taz
TABLES:
- land_use
1 change: 1 addition & 0 deletions configs/resident/settings.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -135,6 +135,7 @@ models:
- compute_disaggregate_accessibility
### mp_initialize_hhs (single process)
- initialize_landuse
- initialize_landuse_taz
- initialize_households
### mp_accessibility
- compute_accessibility
Expand Down
1 change: 1 addition & 0 deletions extensions/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,3 +4,4 @@
from . import transponder_ownership
from . import airport_returns
from . import adjust_auto_operating_cost
from . import annotate_landuse_taz
32 changes: 32 additions & 0 deletions extensions/annotate_landuse_taz.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,32 @@
from __future__ import annotations

import logging

from activitysim.core import chunk, workflow
from activitysim.abm.models.initialize import InitializeTableSettings, annotate_tables

logger = logging.getLogger(__name__)


@workflow.step
def initialize_landuse_taz(
state: workflow.State,
model_settings: InitializeTableSettings | None = None,
model_settings_file_name: str = "initialize_landuse_taz.yaml",
trace_label: str = "initialize_landuse_taz",
) -> None:
"""
Initialize the land use table.
Parameters
----------
state : State
"""
if model_settings is None:
model_settings = InitializeTableSettings.read_settings_file(
state.filesystem,
model_settings_file_name,
mandatory=True,
)
with chunk.chunk_log(state, trace_label, base=True) as chunk_sizer:
annotate_tables(state, model_settings, trace_label, chunk_sizer)

0 comments on commit 106e631

Please sign in to comment.