diff --git a/docs/changes/newsfragments/444.bugfix b/docs/changes/newsfragments/444.bugfix new file mode 100644 index 0000000000..152633f24b --- /dev/null +++ b/docs/changes/newsfragments/444.bugfix @@ -0,0 +1 @@ +Allow ``timeseries_2d`` storage for :class:`.HDF5FeatureStorage` by `Fede Raimondo`_ diff --git a/junifer/storage/hdf5.py b/junifer/storage/hdf5.py index 7a083509ae..99c2f406e8 100644 --- a/junifer/storage/hdf5.py +++ b/junifer/storage/hdf5.py @@ -157,7 +157,13 @@ def __init__( uri.parent.mkdir(parents=True, exist_ok=True) # Available storage kinds - storage_types = ["vector", "timeseries", "matrix", "scalar_table"] + storage_types = [ + "vector", + "timeseries", + "matrix", + "scalar_table", + "timeseries_2d", + ] super().__init__( uri=uri, @@ -180,7 +186,13 @@ def get_valid_inputs(self) -> list[str]: storage. """ - return ["matrix", "vector", "timeseries", "scalar_table"] + return [ + "matrix", + "vector", + "timeseries", + "scalar_table", + "timeseries_2d", + ] def _fetch_correct_uri_for_io(self, element: Optional[dict]) -> str: """Return proper URI for I/O based on `element`. diff --git a/junifer/storage/tests/test_hdf5.py b/junifer/storage/tests/test_hdf5.py index 9f9adfb19c..37e78f023d 100644 --- a/junifer/storage/tests/test_hdf5.py +++ b/junifer/storage/tests/test_hdf5.py @@ -29,6 +29,7 @@ def test_get_valid_inputs() -> None: "vector", "timeseries", "scalar_table", + "timeseries_2d", ]