From 40c7f2ffb57944a9532e4a8ba38165370ca2f879 Mon Sep 17 00:00:00 2001
From: "Alinson S. Xavier" <git@axavier.org>
Date: Fri, 9 Jun 2023 10:57:54 -0500
Subject: [PATCH] io: Simplify more extensions

---
 miplearn/io.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/miplearn/io.py b/miplearn/io.py
index 635be8c..ccad431 100644
--- 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