Fixes the problem described in issue 188.

MattTriano · Oct 22, 2023 · 1c97025 · 1c97025
1 parent 5edb68a
commit 1c97025
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/airflow/dags/run_a_dbt_run_command.py b/airflow/dags/run_a_dbt_run_command.py
@@ -8,7 +8,7 @@
 from airflow.decorators import dag, task
 from airflow.utils.trigger_rule import TriggerRule
 
-from sources.tables import COOK_COUNTY_PARCEL_SALES as SOCRATA_TABLE
+from sources.tables import CHICAGO_CTA_BUS_STOPS as SOCRATA_TABLE
 
 task_logger = logging.getLogger("airflow.task")
 
@@ -27,7 +27,7 @@
 def run_specific_dbt_model_for_a_data_set(table_name: str, task_logger: Logger) -> None:
     dbt_cmd = f"""cd /opt/airflow/dbt && \
                   dbt --warn-error run --select \
-                  re_dbt.dwh.cook_county_parcel_sales_fact+"""
+                  re_dbt.standardized.chicago_cta_bus_stops_standardized+"""
     task_logger.info(f"dbt run command: {dbt_cmd}")
     try:
         subproc_output = subprocess.run(

diff --git a/airflow/dbt/models/standardized/chicago_cta_bus_stops_standardized.sql b/airflow/dbt/models/standardized/chicago_cta_bus_stops_standardized.sql
@@ -4,7 +4,7 @@
 
 WITH records_with_basic_cleaning AS (
     SELECT
-        systemstop::smallint                                  AS systemstop,
+        split_part(systemstop, '.', 1)::smallint              AS systemstop,
         upper(public_nam::text)                               AS public_name,
         upper(dir::text)                                      AS dir,
         upper(street::text)                                   AS street,