Fix lazy inputs support in as_qvalue_or_expr_with_list_to_slice_support

PiperOrigin-RevId: 863576628
Change-Id: Icc7b895de06d7d53f9a56da3c3a1417307cb2e86

Author: apronchenkov

py/koladata/types/py_boxing.py

@@ -167,12 +167,17 @@ def _wrap(x: Any) -> arolla.Expr | arolla.QValue:
   return data_item.DataItem.from_vals(arg)
 
 
+_make_slice_op = arolla.abc.unsafe_make_registered_operator('kd.slices.slice')
+
+
 def as_qvalue_or_expr_with_list_to_slice_support(
     arg: Any,
 ) -> arolla.Expr | arolla.QValue:
-  if isinstance(arg, list):
-    return data_slice.DataSlice.from_vals(arg)
-  return as_qvalue_or_expr(arg)
+  """Converts Python values into QValues or Exprs, supporting list to slice."""
+  if not isinstance(arg, list):
+    return as_qvalue_or_expr(arg)
+  result, _ = arolla.abc.aux_bind_arguments(_make_slice_op, arg)
+  return result
 
 
 def as_qvalue_or_expr_with_py_function_to_py_object_support(

py/koladata/types/py_boxing_test.py

@@ -126,6 +126,7 @@ def test_register_py_function_boxing_fn(self):
       (arolla.L.x, arolla.L.x),
       ((1, 2), arolla.tuple(ds(1), ds(2))),
       ([1, 2], ds([1, 2])),
+      ([[1]], ds([[1]])),
   )
   def test_as_qvalue_or_expr_with_list_to_slice_support(
       self, value, expected_res
@@ -135,6 +136,18 @@ def test_as_qvalue_or_expr_with_list_to_slice_support(
         expected_res,
     )
 
+  def test_as_qvalue_or_expr_with_list_to_slice_exprs(self):
+    testing.assert_equal(
+        arolla.eval(
+            py_boxing.as_qvalue_or_expr_with_list_to_slice_support(
+                [[1], [arolla.L.x], [arolla.L.y]]
+            ),
+            x=ds(2),
+            y=ds(3),
+        ),
+        ds([[1], [2], [3]]),
+    )
+
   def test_as_qvalue_or_expr_with_py_function_to_py_object_support(self):
     fn = lambda x: x
     self.assertIsInstance(