Allow files with prefix .hdf to be processed

callumforrester · Mar 1, 2017 · 9ad90b4 · 9ad90b4
1 parent 85671eb
commit 9ad90b4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/vdsgen/vdsgen.py b/vdsgen/vdsgen.py
@@ -46,7 +46,7 @@ def find_files(path, prefix):
         list: HDF5 files in folder that have the given prefix
 
     """
-    regex = re.compile(prefix + r"\d\.(hdf5|h5)")
+    regex = re.compile(prefix + r"\d+\.(hdf5|hdf|h5)")
 
     files = []
     for file_ in sorted(os.listdir(path)):