From 6a713f717ba3fd53ed2cc770c8d6a6824c9cf618 Mon Sep 17 00:00:00 2001 From: rly Date: Thu, 4 Apr 2024 16:25:35 -0700 Subject: [PATCH] Add comment about hdmf fix --- tests/test_results_table.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tests/test_results_table.py b/tests/test_results_table.py index 165ad04..1aa8042 100644 --- a/tests/test_results_table.py +++ b/tests/test_results_table.py @@ -59,7 +59,9 @@ def test_add_col_dupe_name(self): def test_add_tvt_split(self): rt = ResultsTable(name="foo", description="a test results table") rt.add_tvt_split(np.uint([0, 1, 2, 0, 1])) - assert isinstance(rt["tvt_split"], TrainValidationTestSplit) + # waiting on fix upstream https://github.com/hdmf-dev/hdmf/pull/1091 + # assert isinstance(rt["tvt_split"], TrainValidationTestSplit) + assert isinstance(rt["tvt_split"], EnumData) assert all(rt["tvt_split"].data == np.uint([0, 1, 2, 0, 1])) assert isinstance(rt["tvt_split"].elements, VectorData) assert rt["tvt_split"].elements.data == ["train", "validate", "test"]