Merge pull request #6 from wtysos11/fix_timediff

Fix timediff.py

Author: nnkumar13

timediff.py

@@ -7,7 +7,7 @@
 
 import pyspark
 from pyspark.sql import SparkSession
-from pyspark.sql.functions import col
+from pyspark.sql.functions import col, round
 from pyspark.sql.functions import to_timestamp, current_timestamp
 from pyspark.sql.types import StructType, StructField, StringType, IntegerType, LongType
 
@@ -25,7 +25,7 @@
 
 df.withColumn('input_timestamp',to_timestamp(col('input_timestamp')))\
   .withColumn('current_timestamp', current_timestamp().alias('current_timestamp'))\
-  .withColumn('DiffInSeconds',current_timestamp().cast(LongType) - col('input_timestamp').cast(LongType))\
+  .withColumn('DiffInSeconds',current_timestamp().cast(LongType()) - col('input_timestamp').cast(LongType()))\
   .withColumn('DiffInMinutes',round(col('DiffInSeconds')/60))\
   .withColumn('DiffInHours',round(col('DiffInSeconds')/3600))\
   .withColumn('DiffInDays',round(col('DiffInSeconds')/24*3600))\