replace orca inject with state object

ActivitySim · Mar 7, 2024 · 48be6f4 · 48be6f4
1 parent 4201d36
commit 48be6f4
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/activitysim/abm/models/location_choice.py b/activitysim/abm/models/location_choice.py
@@ -159,7 +159,7 @@ def _location_sample(
         "dest_col_name": skims.dest_key,  # added for sharrow flows
         "timeframe": "timeless",
         "reindex": reindex,
-        "land_use": inject.get_table("land_use").to_frame(),
+        "land_use": state.get_dataframe("land_use"),
     }
     locals_d.update(model_settings.CONSTANTS or {})
 
@@ -624,7 +624,7 @@ def run_location_simulate(
         "dest_col_name": skims.dest_key,  # added for sharrow flows
         "timeframe": "timeless",
         "reindex": reindex,
-        "land_use": inject.get_table("land_use").to_frame(),
+        "land_use": state.get_dataframe("land_use"),
     }
     locals_d.update(model_settings.CONSTANTS or {})
 

diff --git a/activitysim/abm/models/trip_destination.py b/activitysim/abm/models/trip_destination.py
@@ -193,7 +193,7 @@ def _destination_sample(
             "size_terms": size_term_matrix,
             "size_terms_array": size_term_matrix.df.to_numpy(),
             "timeframe": "trip",
-            "land_use": inject.get_table("land_use").to_frame(),
+            "land_use": state.get_dataframe("land_use"),
         }
     )
     locals_dict.update(skims)
@@ -926,7 +926,7 @@ def trip_destination_simulate(
             "size_terms": size_term_matrix,
             "size_terms_array": size_term_matrix.df.to_numpy(),
             "timeframe": "trip",
-            "land_use": inject.get_table("land_use").to_frame(),
+            "land_use": state.get_dataframe("land_use"),
         }
     )
     locals_dict.update(skims)