Skip to content

Commit

Permalink
add drop_unused_columns to compute_settings
Browse files Browse the repository at this point in the history
  • Loading branch information
i-am-sijia committed Apr 19, 2024
1 parent 7fde7b4 commit ef9e9e8
Showing 1 changed file with 8 additions and 0 deletions.
8 changes: 8 additions & 0 deletions activitysim/core/configuration/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -201,6 +201,13 @@ class ComputeSettings(PydanticBase):
for more information.
"""

drop_unused_columns: bool = True
"""Drop unused columns in the choosers df.
Set to True or False to drop unused columns in data table for specific component.
Default to True. If set to False, all columns in the data table will be kept.
"""

def should_skip(self, subcomponent: str) -> bool:
"""Check if sharrow should be skipped for a particular subcomponent."""
if isinstance(self.sharrow_skip, dict):
Expand Down Expand Up @@ -232,6 +239,7 @@ def subcomponent_settings(self, subcomponent: str) -> ComputeSettings:
use_bottleneck=self.use_bottleneck,
use_numexpr=self.use_numexpr,
use_numba=self.use_numba,
drop_unused_columns=self.drop_unused_columns,
)


Expand Down

0 comments on commit ef9e9e8

Please sign in to comment.