From d87476631a7b51afeee8904004839fb41fe9b837 Mon Sep 17 00:00:00 2001
From: Alinson S Xavier <axavier@anl.gov>
Date: Tue, 31 Aug 2021 14:15:22 -0500
Subject: [PATCH] Store ML training data in a separate HDF5 file

---
 src/instance/file_instance.jl       | 2 +-
 test/instance/file_instance_test.jl | 2 +-
 test/instance/jump_instance_test.jl | 2 +-
 3 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/instance/file_instance.jl b/src/instance/file_instance.jl
index 7e296df..e7c7915 100644
--- a/src/instance/file_instance.jl
+++ b/src/instance/file_instance.jl
@@ -15,7 +15,7 @@ mutable struct FileInstance <: Instance
     function FileInstance(filename::AbstractString, build_model::Function)::FileInstance
         instance = new(nothing, nothing, filename, nothing, build_model)
         instance.py = PyFileInstance(instance)
-        instance.h5 = Hdf5Sample(filename)
+        instance.h5 = Hdf5Sample("$filename.h5", mode="a")
         instance.filename = filename
         return instance
     end
diff --git a/test/instance/file_instance_test.jl b/test/instance/file_instance_test.jl
index 32afe80..182e4ec 100644
--- a/test/instance/file_instance_test.jl
+++ b/test/instance/file_instance_test.jl
@@ -15,7 +15,7 @@ using Cbc
         solver = LearningSolver(Cbc.Optimizer)
         solve!(solver, instance)
 
-        h5 = Hdf5Sample(filename)
+        h5 = Hdf5Sample("$filename.h5")
         @test h5.get_scalar("mip_wallclock_time") > 0
     end
 
diff --git a/test/instance/jump_instance_test.jl b/test/instance/jump_instance_test.jl
index afcbf38..c5851d1 100644
--- a/test/instance/jump_instance_test.jl
+++ b/test/instance/jump_instance_test.jl
@@ -56,7 +56,7 @@ end
         instance = FileInstance(filename, build_model)
         solver = LearningSolver(Cbc.Optimizer)
         solve!(solver, instance)
-        h5 = MIPLearn.Hdf5Sample(filename)
+        h5 = MIPLearn.Hdf5Sample("$filename.h5")
         @test h5.get_array("mip_var_values") == [1.0, 0.0]
     end
 end