review improvements by mwestphal

openmc-dev · Jan 9, 2025 · aa13835 · aa13835
1 parent b2c775c
commit aa13835
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/openmc/mesh.py b/openmc/mesh.py
@@ -2369,7 +2369,7 @@ def write_data_to_vtk(
         Parameters
         ----------
         filename : str or pathlib.Path
-            Name of the VTK file to write. If the filename ends in '.hdf' then a VTKHDF
+            Name of the VTK file to write. If the filename ends in '.vtkhdf' then a VTKHDF
             format file will be written and can be opened with Paraview versions 5.13.0
             and above, if the filename ends in '.vtk' then a .vtk file will be written.
         datasets : dict
@@ -2381,7 +2381,7 @@ def write_data_to_vtk(
             volume of the mesh elements
         """
 
-        if Path(filename).suffix == ".hdf":
+        if Path(filename).suffix == ".vtkhdf":
 
             def append_dataset(dset, array):
                 """Convenience function to append data to an HDF5 dataset"""
@@ -2558,7 +2558,7 @@ def append_dataset(dset, array):
 
         else:
             raise ValueError(
-                f"Unsupported file extension for '{filename}'. Extension must be '.hdf' or '.vtk'."
+                f"Unsupported file extension for '{filename}'. Extension must be '.vtkhdf' or '.vtk'."
             )
 
     @classmethod