From 48767f8477d05037790318fc0750b3fd8ec8b678 Mon Sep 17 00:00:00 2001 From: Adam Su Date: Fri, 29 May 2026 21:48:10 -0400 Subject: [PATCH] updated citibike_utils --- src/citibike/citibike_utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/citibike/citibike_utils.py b/src/citibike/citibike_utils.py index 99b5530..ee2e631 100644 --- a/src/citibike/citibike_utils.py +++ b/src/citibike/citibike_utils.py @@ -1,8 +1,8 @@ -from pyspark.sql.functions import unix_timestamp, col, round +from pyspark.sql.functions import unix_timestamp, round def get_trip_duration_mins(spark, df, start_col, end_col, output_col): return df.withColumn( output_col, - round((unix_timestamp(col(end_col)) - unix_timestamp(col(start_col))) / 60, 2) + round((unix_timestamp(end_col) - unix_timestamp(start_col)) / 60, 2) ) \ No newline at end of file