def test_concat(self):
        """Test that concat is correctly computed."""
        t0 = torch.zeros(10)
        t1 = torch.ones(10)
        da = DataAttribute(torch.cat([t0, t1]), "logits")
        self.assertEqual(list(da.concat(da).data), list(torch.cat([t0, t1, t0, t1])))
Test that concat is correctly computed.
tests/benchmarks/utils/test_data_attribute.py