remove dupe name column in gtfs_utils_v2

cal-itp · Jun 7, 2024 · 4f23318 · 4f23318
1 parent bbb1d57
commit 4f23318
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/_shared_utils/shared_utils/gtfs_utils_v2.py b/_shared_utils/shared_utils/gtfs_utils_v2.py
@@ -252,14 +252,14 @@ def schedule_daily_feed_to_gtfs_dataset_name(
     # Get GTFS schedule datasets from Airtable
     dim_gtfs_datasets = schedule_rt_utils.filter_dim_gtfs_datasets(
         keep_cols=["key", "name", "type", "regional_feed_type"], custom_filtering={"type": ["schedule"]}, get_df=False
-    ) >> rename(name="gtfs_dataset_name")
+    )
 
     # Merge on gtfs_dataset_key to get organization name
     fact_feeds = (
         tbls.mart_gtfs.fct_daily_schedule_feeds()
         >> filter(_.date == selected_date)
-        >> inner_join(_, dim_gtfs_datasets, on="gtfs_dataset_key")
-    )
+        >> inner_join(_, dim_gtfs_datasets, on=["gtfs_dataset_key", "gtfs_dataset_name"])
+    ) >> rename(name="gtfs_dataset_name")
 
     if get_df:
         fact_feeds = (