fix resource closed error

mbta · Dec 12, 2023 · f2c301a · f2c301a
1 parent 7c23155
commit f2c301a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python_src/src/lamp_py/tableau/jobs/rt_rail.py b/python_src/src/lamp_py/tableau/jobs/rt_rail.py
@@ -95,7 +95,6 @@ def __init__(self) -> None:
             " ORDER BY "
             "   ve.service_date, vt.route_id, vt.direction_id, vt.vehicle_id"
             ";"
-            "SET SESSION timezone to 'UTC';"
         )
 
     @property
@@ -185,6 +184,7 @@ def update_parquet(self, db_manager: DatabaseManager) -> bool:
             write_path=db_parquet_path,
             schema=self.parquet_schema,
         )
+        db_manager.execute(sa.text("SET SESSION timezone to 'UTC';"))
 
         # update downloaded parquet file with filtered service_date
         old_filter = pc.field("service_date") < max_start_date