Add return a value in main function

chihacknight · Feb 12, 2024 · 78b9028 · 78b9028
1 parent 8eeb097
commit 78b9028
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/data_analysis/ridership_to_json.py b/data_analysis/ridership_to_json.py
@@ -138,7 +138,7 @@ def main(month: int = None, year: int = None, save: bool = True) -> None:
         'jyb9-n7fm/rows.csv?accessType=DOWNLOAD'
     )
     print("Done!")
-    ridership_to_json(ridership_df=ridership_df, month=month, year=year, save=save)
+    return ridership_to_json(ridership_df=ridership_df, month=month, year=year, save=save)
 
 
 if __name__ == '__main__':