Create html dir if doens't exist.

InstituteforDiseaseModeling · Jun 19, 2024 · b39d320 · b39d320
1 parent aa650cb
commit b39d320
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/jb/utils/folium_animate_from_sqlite.py b/jb/utils/folium_animate_from_sqlite.py
@@ -4,6 +4,7 @@
 import numpy as np
 import sys
 import argparse
+import os
 
 import folium
 from folium.plugins import HeatMapWithTime
@@ -98,6 +99,8 @@ def renorm( value ):
     heat_map.add_to(m)
 
     # Save the map as an HTML file
+    if not os.path.exists( "html" ):
+        os.mkdir( "html" )
     m.save( output_file )
 
     print( f"File written: {output_file}" )