Integrate kd.types.DataItem with the ipython extension arolla.experimental.colab_safe_repr

This change list adds the attribute `DataItem._COLAB_HAS_SAFE_REPR = True` to fulfill the interface requirements for `arolla.experimental.colab_safe_repr`.

PiperOrigin-RevId: 865144620
Change-Id: I6a1024b6383acbb33fb17a36383e90e098662e3b

Author: apronchenkov

py/koladata/types/data_item.cc

@@ -189,6 +189,7 @@ PyTypeObject* InitPyDataItemType() {
   };
 
   PyObject* qvalue_subtype = PyType_FromSpec(&spec);
+  PyObject_SetAttrString(qvalue_subtype, "_COLAB_HAS_SAFE_REPR", Py_True);
   if (!arolla::python::RegisterPyQValueSpecializationByKey(
           kDataItemQValueSpecializationKey, qvalue_subtype)) {
     return nullptr;

py/koladata/types/data_item_test.py

@@ -214,9 +214,7 @@ def test_int(self):
     self.assertAlmostEqual(int(ds(b'42')), 42)
     with self.assertRaisesRegex(ValueError, 'invalid literal for int'):
       int(ds('xyz'))
-    with self.assertRaisesRegex(
-        TypeError, r'must be a .*, not \'NoneType\''
-    ):
+    with self.assertRaisesRegex(TypeError, r'must be a .*, not \'NoneType\''):
       int(ds(None))
     with self.assertRaisesRegex(
         ValueError, 'cannot be a DataItem that holds an ItemId'
@@ -234,7 +232,7 @@ def test_float(self):
     ):
       float(ds(b'xyz'))
     with self.assertRaisesRegex(
-        TypeError, 'must be a string or a real number, not \'NoneType\''
+        TypeError, "must be a string or a real number, not 'NoneType'"
     ):
       float(ds(None))
     with self.assertRaisesRegex(
@@ -326,6 +324,9 @@ def test_signatures(self):
       if callable(fn):
         _ = inspect.signature(fn)  # Shouldn't raise.
 
+  def test_colab_has_safe_repr(self):
+    self.assertTrue(hasattr(data_item.DataItem, '_COLAB_HAS_SAFE_REPR'))
+
 
 if __name__ == '__main__':
   absltest.main()

py/koladata/types/data_slice_test.py

@@ -4443,6 +4443,8 @@ def test_functor_repr(self, f, expected_substrings):
     for line in expected_substrings:
       self.assertIn(line, fn_repr)
 
+  def test_colab_has_safe_repr(self):
+    self.assertFalse(hasattr(data_slice.DataSlice, '_COLAB_HAS_SAFE_REPR'))
 
 if __name__ == '__main__':
   absltest.main()