update test on label shape

dmlc · Mar 22, 2023 · 8aafc35 · 8aafc35
1 parent 4b9ac2a
commit 8aafc35
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/python/common/data/test_data.py b/tests/python/common/data/test_data.py
@@ -429,7 +429,8 @@ def test_zinc():
         g2, _ = dataset2[0]
 
         assert g2.num_edges() - g1.num_edges() == g1.num_nodes()
-        assert label.shape[0] == 1
+        # return a scalar tensor
+        assert not label.shape
 
 
 @unittest.skipIf(