Deprecate the kd.map_py_on_present operator

PiperOrigin-RevId: 715043659
Change-Id: I145924c618c5d545df93f56b7c8f53bd3ce19530

Author: apronchenkov

py/koladata/operators/py.py

@@ -18,6 +18,7 @@
 import functools
 import itertools
 from typing import Any, Callable, Iterable
+import warnings
 
 from arolla import arolla
 from koladata.expr import py_expr_eval_py_ext
@@ -822,6 +823,11 @@ def map_py_on_present(
   Returns:
     Result DataSlice.
   """
+  warnings.warn(
+      '`kd.map_py_on_present` is deprecated and will be removed, please use'
+      ' `kd.map_py` instead',
+      DeprecationWarning,
+  )
   if not args and not kwargs:
     raise TypeError('expected at least one input DataSlice, got none')
   cond = functools.reduce(

py/koladata/operators/tests/py_map_py_on_present_test.py

@@ -28,12 +28,16 @@
 kde = kde_operators.kde
 
 
-class PyMapPyOnSelectedTest(parameterized.TestCase):
+class PyMapPyOnPresentTest(parameterized.TestCase):
 
   # Note: This operator is assembled from the same building blocks as
   # the operator `kde.py.map_py`, and these are tested together with that
   # operator.
 
+  def test_deprecated(self):
+    with self.assertWarns(DeprecationWarning):
+      _ = expr_eval.eval(kde.py.map_py_on_present(lambda x: x, ds([])))
+
   def test_args_kwargs(self):
     x = ds([1, 2, None])
     y = ds([3, None, 4])