Add comment about hdmf fix

hdmf-dev · Apr 4, 2024 · 6a713f7 · 6a713f7
1 parent 8ffe6fe
commit 6a713f7
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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"]