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"]