Rename "experimental_aux_policy" to "aux_policy"

Motivation: The aux-policy mechanism is no longer considered experimental. While its usage patterns may continue to evolve, the core api is unlikely to undergo major changes.
PiperOrigin-RevId: 861470059
Change-Id: I3d5ad6676664f4ce258264bd09f7486a99505570

Author: apronchenkov

py/koladata/operators/bags.py

@@ -62,7 +62,7 @@ def is_null_bag(bag):  # pylint: disable=unused-argument
         _qtype_utils.expect_data_bag_args(_P.bags),
     ],
     qtype_inference_expr=_qtypes.DATA_BAG,
-    experimental_aux_policy=_aux_policies.CLASSIC_AUX_POLICY,
+    aux_policy=_aux_policies.CLASSIC_AUX_POLICY,
 )
 def enriched(*bags):  # pylint: disable=unused-argument
   """Creates a new immutable DataBag enriched by `bags`.
@@ -93,7 +93,7 @@ def enriched(*bags):  # pylint: disable=unused-argument
         _qtype_utils.expect_data_bag_args(_P.bags),
     ],
     qtype_inference_expr=_qtypes.DATA_BAG,
-    experimental_aux_policy=_aux_policies.CLASSIC_AUX_POLICY,
+    aux_policy=_aux_policies.CLASSIC_AUX_POLICY,
 )
 def updated(*bags):  # pylint: disable=unused-argument
   """Creates a new immutable DataBag updated by `bags`.

py/koladata/operators/classic_binding_policy_test.py

@@ -31,8 +31,7 @@
 
 
 @arolla.optools.as_lambda_operator(
-    'op_with_classic_binding_policy',
-    experimental_aux_policy=aux_policies.CLASSIC_AUX_POLICY,
+    'op_with_classic_binding_policy', aux_policy=aux_policies.CLASSIC_AUX_POLICY
 )
 def op_with_classic_binding_policy(x, y):
   return (x, y)

py/koladata/operators/core.py

@@ -343,7 +343,7 @@ def stub(x, attrs=data_slice.DataSlice.from_vals([])):  # pylint: disable=unused
         qtype_utils.expect_data_bag_args(P.bag),
     ],
     qtype_inference_expr=qtypes.DATA_SLICE,
-    experimental_aux_policy=aux_policies.CLASSIC_AUX_POLICY,
+    aux_policy=aux_policies.CLASSIC_AUX_POLICY,
 )
 def updated(ds, *bag):  # pylint: disable=unused-argument
   """Returns a copy of a DataSlice with DataBag(s) of updates applied.
@@ -1266,7 +1266,7 @@ def with_merged_bag(ds):  # pylint: disable=unused-argument
         qtype_utils.expect_data_bag_args(P.bag),
     ],
     qtype_inference_expr=qtypes.DATA_SLICE,
-    experimental_aux_policy=aux_policies.CLASSIC_AUX_POLICY,
+    aux_policy=aux_policies.CLASSIC_AUX_POLICY,
 )
 def enriched(ds, *bag):  # pylint: disable=unused-argument
   """Returns a copy of a DataSlice with a additional fallback DataBag(s).

py/koladata/operators/dicts.py

@@ -51,7 +51,7 @@ def _shaped(
 )
 @arolla.optools.as_lambda_operator(
     'kd.dicts.new',
-    experimental_aux_policy='koladata_adhoc_binding_policy[kd.dicts.new]',
+    aux_policy='koladata_adhoc_binding_policy[kd.dicts.new]',
     qtype_constraints=[
         qtype_utils.expect_data_slice_or_unspecified(P.keys),
         qtype_utils.expect_data_slice_or_unspecified(P.values),

py/koladata/operators/lists.py

@@ -294,7 +294,7 @@ def _new(items, item_schema, schema, itemid):  # pylint: disable=unused-argument
 )
 @arolla.optools.as_lambda_operator(
     'kd.lists.new',
-    experimental_aux_policy='koladata_adhoc_binding_policy[kd.lists.new]',
+    aux_policy='koladata_adhoc_binding_policy[kd.lists.new]',
     qtype_constraints=[
         qtype_utils.expect_data_slice_or_unspecified(P.items),
         qtype_utils.expect_data_slice_or_unspecified(P.item_schema),

py/koladata/operators/optools.py

@@ -248,7 +248,7 @@ def impl(fn) -> arolla.types.RegisteredOperator:
       overloadable_op = arolla.optools.add_to_registry_as_overloadable(
           name,
           if_present='unsafe_override' if unsafe_override else 'raise',
-          experimental_aux_policy=aux_policy,
+          aux_policy=aux_policy,
       )(fn)
     arolla.abc.set_expr_view_for_registered_operator(name, view)
     arolla.abc.register_op_repr_fn_by_registration_name(name, repr_fn)

py/koladata/operators/slices.py

@@ -242,7 +242,7 @@ def size(x):
         qtype_utils.expect_data_slice_args(P.args),
     ],
     qtype_inference_expr=P.args,
-    experimental_aux_policy=aux_policies.CLASSIC_AUX_POLICY,
+    aux_policy=aux_policies.CLASSIC_AUX_POLICY,
 )
 def align(*args):  # pylint: disable=unused-argument
   """Expands all of the DataSlices in `args` to the same common shape.
@@ -1566,7 +1566,7 @@ def _expect_data_slices_or_slices_or_ellipsis(value):
         _expect_data_slices_or_slices_or_ellipsis(P.slices),
     ],
     qtype_inference_expr=qtypes.DATA_SLICE,
-    experimental_aux_policy=aux_policies.CLASSIC_AUX_POLICY,
+    aux_policy=aux_policies.CLASSIC_AUX_POLICY,
 )
 def subslice(x, *slices):  # pylint: disable=unused-argument
   """Slices `x` across all of its dimensions based on the provided `slices`.
@@ -1914,7 +1914,7 @@ def _zip(*args):
 @optools.add_to_registry(aliases=['kd.item'], via_cc_operator_package=True)
 @arolla.optools.as_lambda_operator(
     'kd.slices.item',
-    experimental_aux_policy='koladata_adhoc_binding_policy[kd.slices.item]',
+    aux_policy='koladata_adhoc_binding_policy[kd.slices.item]',
     qtype_constraints=[
         qtype_utils.expect_data_slice(P.x),
         qtype_utils.expect_data_slice_or_unspecified(P.schema),
@@ -1984,7 +1984,7 @@ def _item_bind_args(x, schema=arolla.unspecified()):
 @optools.add_to_registry(aliases=['kd.slice'], via_cc_operator_package=True)
 @arolla.optools.as_lambda_operator(
     'kd.slices.slice',
-    experimental_aux_policy='koladata_adhoc_binding_policy[kd.slices.slice]',
+    aux_policy='koladata_adhoc_binding_policy[kd.slices.slice]',
     qtype_constraints=[
         qtype_utils.expect_data_slice(P.x),
         qtype_utils.expect_data_slice_or_unspecified(P.schema),
@@ -2067,7 +2067,7 @@ def _typed_slice_bind_args(x, schema):
 @optools.add_to_registry(aliases=['kd.int32'], via_cc_operator_package=True)
 @arolla.optools.as_lambda_operator(
     'kd.slices.int32',
-    experimental_aux_policy='koladata_adhoc_binding_policy[kd.slices.int32]',
+    aux_policy='koladata_adhoc_binding_policy[kd.slices.int32]',
     qtype_constraints=[
         qtype_utils.expect_data_slice(P.x),
     ],
@@ -2087,7 +2087,7 @@ def int32(x):
 @optools.add_to_registry(aliases=['kd.int64'], via_cc_operator_package=True)
 @arolla.optools.as_lambda_operator(
     'kd.slices.int64',
-    experimental_aux_policy='koladata_adhoc_binding_policy[kd.slices.int64]',
+    aux_policy='koladata_adhoc_binding_policy[kd.slices.int64]',
     qtype_constraints=[
         qtype_utils.expect_data_slice(P.x),
     ],
@@ -2107,7 +2107,7 @@ def int64(x):
 @optools.add_to_registry(aliases=['kd.float32'], via_cc_operator_package=True)
 @arolla.optools.as_lambda_operator(
     'kd.slices.float32',
-    experimental_aux_policy='koladata_adhoc_binding_policy[kd.slices.float32]',
+    aux_policy='koladata_adhoc_binding_policy[kd.slices.float32]',
     qtype_constraints=[
         qtype_utils.expect_data_slice(P.x),
     ],
@@ -2127,7 +2127,7 @@ def float32(x):
 @optools.add_to_registry(aliases=['kd.float64'], via_cc_operator_package=True)
 @arolla.optools.as_lambda_operator(
     'kd.slices.float64',
-    experimental_aux_policy='koladata_adhoc_binding_policy[kd.slices.float64]',
+    aux_policy='koladata_adhoc_binding_policy[kd.slices.float64]',
     qtype_constraints=[
         qtype_utils.expect_data_slice(P.x),
     ],
@@ -2147,7 +2147,7 @@ def float64(x):
 @optools.add_to_registry(aliases=['kd.str'], via_cc_operator_package=True)
 @arolla.optools.as_lambda_operator(
     'kd.slices.str',
-    experimental_aux_policy='koladata_adhoc_binding_policy[kd.slices.str]',
+    aux_policy='koladata_adhoc_binding_policy[kd.slices.str]',
     qtype_constraints=[
         qtype_utils.expect_data_slice(P.x),
     ],
@@ -2227,7 +2227,7 @@ def get_repr(
 @optools.add_to_registry(aliases=['kd.bytes'], via_cc_operator_package=True)
 @arolla.optools.as_lambda_operator(
     'kd.slices.bytes',
-    experimental_aux_policy='koladata_adhoc_binding_policy[kd.slices.bytes]',
+    aux_policy='koladata_adhoc_binding_policy[kd.slices.bytes]',
     qtype_constraints=[
         qtype_utils.expect_data_slice(P.x),
     ],
@@ -2247,7 +2247,7 @@ def bytes_(x):
 @optools.add_to_registry(aliases=['kd.bool'], via_cc_operator_package=True)
 @arolla.optools.as_lambda_operator(
     'kd.slices.bool',
-    experimental_aux_policy='koladata_adhoc_binding_policy[kd.slices.bool]',
+    aux_policy='koladata_adhoc_binding_policy[kd.slices.bool]',
     qtype_constraints=[
         qtype_utils.expect_data_slice(P.x),
     ],
@@ -2267,7 +2267,7 @@ def bool_(x):
 @optools.add_to_registry(aliases=['kd.mask'], via_cc_operator_package=True)
 @arolla.optools.as_lambda_operator(
     'kd.slices.mask',
-    experimental_aux_policy='koladata_adhoc_binding_policy[kd.slices.mask]',
+    aux_policy='koladata_adhoc_binding_policy[kd.slices.mask]',
     qtype_constraints=[
         qtype_utils.expect_data_slice(P.x),
     ],
@@ -2289,7 +2289,7 @@ def mask(x):
 )
 @arolla.optools.as_lambda_operator(
     'kd.slices.expr_quote',
-    experimental_aux_policy=(
+    aux_policy=(
         'koladata_adhoc_binding_policy[kd.slices.expr_quote]'
     ),
     qtype_constraints=[

py/koladata/operators/strings.py

@@ -170,7 +170,7 @@ def find(
         qtype_utils.expect_data_slice_args(P.args),
     ],
     qtype_inference_expr=qtypes.DATA_SLICE,
-    experimental_aux_policy=aux_policies.CLASSIC_AUX_POLICY,
+    aux_policy=aux_policies.CLASSIC_AUX_POLICY,
 )
 def printf(fmt, *args):  # pylint: disable=unused-argument
   """Formats strings according to printf-style (C++) format strings.
@@ -244,10 +244,8 @@ def format_(fmt, /, **kwargs):  # pylint: disable=unused-argument
 @optools.add_to_registry(aliases=['kd.fstr'], via_cc_operator_package=True)
 @arolla.optools.as_lambda_operator(
     'kd.strings.fstr',
-    experimental_aux_policy='koladata_adhoc_binding_policy[kd.strings.fstr]',
-    qtype_constraints=[
-        qtype_utils.expect_data_slice(P.fmt),
-    ],
+    aux_policy='koladata_adhoc_binding_policy[kd.strings.fstr]',
+    qtype_constraints=[qtype_utils.expect_data_slice(P.fmt)],
 )
 def fstr(fmt):
   """Transforms Koda f-string into an expression.
@@ -329,7 +327,7 @@ def _fstr_bind_args(fstr, /):  # pylint: disable=redefined-outer-name
         qtype_utils.expect_data_slice_args(P.args),
     ],
     qtype_inference_expr=qtypes.DATA_SLICE,
-    experimental_aux_policy=aux_policies.CLASSIC_AUX_POLICY,
+    aux_policy=aux_policies.CLASSIC_AUX_POLICY,
 )
 def _test_only_format_wrapper(fmt, arg_names, *args):  # pylint: disable=unused-argument
   """Test only wrapper for format with Arolla signature."""
@@ -347,7 +345,7 @@ def _test_only_format_wrapper(fmt, arg_names, *args):  # pylint: disable=unused-
         qtype_utils.expect_data_slice_args(P.args),
     ],
     qtype_inference_expr=qtypes.DATA_SLICE,
-    experimental_aux_policy=aux_policies.CLASSIC_AUX_POLICY,
+    aux_policy=aux_policies.CLASSIC_AUX_POLICY,
 )
 def join(*args):  # pylint: disable=unused-argument
   """Concatenates the given strings.

py/koladata/operators/tuple.py

@@ -33,8 +33,7 @@
 # implemented Koladata-specific boxing rules.
 @optools.add_to_registry(aliases=['kd.tuple'], via_cc_operator_package=True)
 @arolla.optools.as_lambda_operator(
-    'kd.tuples.tuple',
-    experimental_aux_policy=aux_policies.CLASSIC_AUX_POLICY,
+    'kd.tuples.tuple', aux_policy=aux_policies.CLASSIC_AUX_POLICY
 )
 def tuple_(*args):
   """Returns a tuple-like object containing the given `*args`."""