check if is directory befoe mkdirs (#827)

Author: HydrogenSulfate

ppsci/solver/solver.py

@@ -744,7 +744,8 @@ def export(
         )
 
         # save static graph model to disk
-        os.makedirs(osp.dirname(export_path), exist_ok=True)
+        if osp.isdir(osp.dirname(export_path)):
+            os.makedirs(osp.dirname(export_path), exist_ok=True)
         try:
             jit.save(static_model, export_path)
         except Exception as e:

ppsci/utils/logger.py

@@ -104,7 +104,8 @@ def init_logger(
     # add file_handler, output to log_file(if specified), only for rank 0 device
     if log_file is not None and dist.get_rank() == 0:
         log_file_folder = os.path.dirname(log_file)
-        os.makedirs(log_file_folder, exist_ok=True)
+        if os.path.isdir(log_file_folder):
+            os.makedirs(log_file_folder, exist_ok=True)
         file_formatter = logging.Formatter(
             "[%(asctime)s] %(name)s %(levelname)s: %(message)s",
             datefmt="%Y/%m/%d %H:%M:%S",

ppsci/utils/symbolic.py

@@ -615,7 +615,8 @@ def add_edge(u: str, v: str, u_color: str = C_DATA, v_color: str = C_DATA):
     graph.layout()
     image_path = f"{graph_filename}.png"
     dot_path = f"{graph_filename}.dot"
-    os.makedirs(os.path.dirname(image_path), exist_ok=True)
+    if os.path.isdir(os.path.dirname(image_path)):
+        os.makedirs(os.path.dirname(image_path), exist_ok=True)
     graph.draw(image_path, prog="dot")
     graph.write(dot_path)
     logger.message(

ppsci/utils/writer.py

@@ -182,7 +182,8 @@ def save_tecplot_file(
     nx, ny = num_x, num_y
     assert nx * ny == nxy, f"nx({nx}) * ny({ny}) != nxy({nxy})"
 
-    os.makedirs(os.path.dirname(filename), exist_ok=True)
+    if os.path.isdir(os.path.dirname(filename)):
+        os.makedirs(os.path.dirname(filename), exist_ok=True)
 
     if filename.endswith(".dat"):
         filename = filename[:-4]

ppsci/visualize/plot.py

@@ -80,7 +80,8 @@ def _save_plot_from_1d_array(filename, coord, value, value_keys, num_timestamps=
         value_keys (Tuple[str, ...]): Value keys.
         num_timestamps (int, optional): Number of timestamps coord/value contains. Defaults to 1.
     """
-    os.makedirs(os.path.dirname(filename), exist_ok=True)
+    if os.path.isdir(os.path.dirname(filename)):
+        os.makedirs(os.path.dirname(filename), exist_ok=True)
     fig, a = plt.subplots(len(value_keys), num_timestamps, squeeze=False)
     fig.subplots_adjust(hspace=0.8)
 
@@ -187,7 +188,8 @@ def _save_plot_from_2d_array(
         xticks (Optional[Tuple[float, ...]]): Tuple of xtick locations. Defaults to None.
         yticks (Optional[Tuple[float, ...]]): Tuple of ytick locations. Defaults to None.
     """
-    os.makedirs(os.path.dirname(filename), exist_ok=True)
+    if os.path.isdir(os.path.dirname(filename)):
+        os.makedirs(os.path.dirname(filename), exist_ok=True)
 
     plt.close("all")
     matplotlib.rcParams["xtick.labelsize"] = 5
@@ -334,7 +336,8 @@ def _save_plot_from_3d_array(
         visu_keys (Tuple[str, ...]): Keys for visualizing data. such as ("u", "v").
         num_timestamps (int, optional): Number of timestamps coord/value contains. Defaults to 1.
     """
-    os.makedirs(os.path.dirname(filename), exist_ok=True)
+    if os.path.isdir(os.path.dirname(filename)):
+        os.makedirs(os.path.dirname(filename), exist_ok=True)
 
     fig = plt.figure(figsize=(10, 10))
     len_ts = len(visu_data[0]) // num_timestamps
@@ -479,7 +482,8 @@ def plot_weather(
             )
         plt.colorbar(mappable=map_, cax=None, ax=None, shrink=0.5, label=colorbar_label)
 
-    os.makedirs(os.path.dirname(filename), exist_ok=True)
+    if os.path.isdir(os.path.dirname(filename)):
+        os.makedirs(os.path.dirname(filename), exist_ok=True)
     fig = plt.figure(facecolor="w", figsize=(7, 7))
     ax = fig.add_subplot(2, 1, 1)
     plot_weather(

ppsci/visualize/vtu.py

@@ -53,7 +53,8 @@ def _save_vtu_from_array(filename, coord, value, value_keys, num_timestamps=1):
     if coord.shape[1] not in [2, 3]:
         raise ValueError(f"ndim of coord({coord.shape[1]}) should be 2 or 3.")
 
-    os.makedirs(os.path.dirname(filename), exist_ok=True)
+    if os.path.isdir(os.path.dirname(filename)):
+        os.makedirs(os.path.dirname(filename), exist_ok=True)
 
     # discard extension name
     if filename.endswith(".vtu"):
@@ -183,5 +184,6 @@ def save_vtu_to_mesh(
         points=points.T, cells=[("vertex", np.arange(npoint).reshape(npoint, 1))]
     )
     mesh.point_data = {key: data_dict[key] for key in value_keys}
-    os.makedirs(os.path.dirname(filename), exist_ok=True)
+    if os.path.isdir(os.path.dirname(filename)):
+        os.makedirs(os.path.dirname(filename), exist_ok=True)
     mesh.write(filename)