From 19e1f52b4fba7a62627c3b086e513492d1cee50f Mon Sep 17 00:00:00 2001 From: "Alinson S. Xavier" Date: Thu, 12 Jun 2025 11:15:09 -0500 Subject: [PATCH] BasicCollector: store `data_filename` in HDF5 file --- miplearn/collectors/basic.py | 1 + 1 file changed, 1 insertion(+) diff --git a/miplearn/collectors/basic.py b/miplearn/collectors/basic.py index a5dac1b..4b65cf2 100644 --- a/miplearn/collectors/basic.py +++ b/miplearn/collectors/basic.py @@ -50,6 +50,7 @@ class BasicCollector: return with H5File(h5_filename, "w") as h5: + h5.put_scalar("data_filename", data_filename) streams: List[Any] = [StringIO()] if verbose: streams += [sys.stdout]