def test_tensor_uniques(self):
        """Test that uniques are correctly computed for tensors."""
        t = torch.zeros(10)
        da = DataAttribute(t, "task_labels")
        self.assertEqual(da.uniques, {0.0})
Test that uniques are correctly computed for tensors.
tests/benchmarks/utils/test_data_attribute.py