io: Simplify more extensions

2 years ago · 40c7f2ffb5
parent 25728f5512
commit 40c7f2ffb5
1 changed files with 2 additions and 1 deletions
--- a/miplearn/io.py
+++ b/miplearn/io.py
@ -86,7 +86,8 @@ def read_pkl_gz(filename: str) -> Any:

 def _to_h5_filename(data_filename: str) -> str:
    output = f"{data_filename}.h5"
-    output = output.replace(".pkl.gz.h5", ".h5")
+    output = output.replace(".gz.h5", ".h5")
+    output = output.replace(".json.h5", ".h5")
    output = output.replace(".pkl.h5", ".h5")
    output = output.replace(".jld2.h5", ".h5")
    return output