format

Author: edasmalchi

_shared_utils/shared_utils/rt_utils.py

@@ -828,8 +828,12 @@ def get_operators(analysis_date, operator_list, verbose=False):
 
 
 def spa_map_export_link(
-    gdf: gpd.GeoDataFrame, path: str, state: dict,
-    site: str = SPA_MAP_SITE, cache_seconds: int = 3600, verbose: bool = False
+    gdf: gpd.GeoDataFrame,
+    path: str,
+    state: dict,
+    site: str = SPA_MAP_SITE,
+    cache_seconds: int = 3600,
+    verbose: bool = False,
 ):
     """
     Called via set_state_export. Handles stream writing of gzipped geojson to GCS bucket,
@@ -855,9 +859,14 @@ def set_state_export(
     bucket: str = SPA_MAP_BUCKET,
     subfolder: str = "testing/",
     filename: str = "test2",
-    map_type: Literal["speedmap", "speed_variation",
-                      "new_speedmap", "new_speed_variation"
-                      "hqta_areas", "hqta_stops", "state_highway_network"] = None,
+    map_type: Literal[
+        "speedmap",
+        "speed_variation",
+        "new_speedmap",
+        "new_speed_variation" "hqta_areas",
+        "hqta_stops",
+        "state_highway_network",
+    ] = None,
     map_title: str = "Map",
     cmap: branca.colormap.ColorMap = None,
     color_col: str = None,