Skip to content

Commit a6504b7

Browse files
felbrocopybara-github
authored andcommitted
Add kd.make_namedtuple as an alias for kd.tuple.make_namedtuple
PiperOrigin-RevId: 721392110 Change-Id: I6d84cdf8cb2d133a2cc1f3f015b694803e322a60
1 parent 134cc12 commit a6504b7

File tree

3 files changed

+8
-1
lines changed

3 files changed

+8
-1
lines changed

py/koladata/operators/tests/BUILD

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7154,6 +7154,7 @@ py_test(
71547154
"//py/koladata/expr:input_container",
71557155
"//py/koladata/expr:view",
71567156
"//py/koladata/operators:kde_operators",
7157+
"//py/koladata/operators:optools",
71577158
"//py/koladata/testing",
71587159
"//py/koladata/types:data_slice",
71597160
"//py/koladata/types:qtypes",

py/koladata/operators/tests/tuple_make_namedtuple_test.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@
1919
from koladata.expr import input_container
2020
from koladata.expr import view
2121
from koladata.operators import kde_operators
22+
from koladata.operators import optools
2223
from koladata.testing import testing
2324
from koladata.types import data_slice
2425
from koladata.types import qtypes
@@ -70,6 +71,11 @@ def test_view(self):
7071
x_namedtuple = kde.tuple.make_namedtuple(x=I.x, y=I.y)
7172
self.assertTrue(view.has_koda_view(x_namedtuple))
7273

74+
def test_alias(self):
75+
self.assertTrue(
76+
optools.equiv_to_op(kde.tuple.make_namedtuple, kde.make_namedtuple)
77+
)
78+
7379

7480
if __name__ == '__main__':
7581
absltest.main()

py/koladata/operators/tuple.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -60,7 +60,7 @@ def make_slice(
6060
return arolla.M.core.make_slice(start, stop, step)
6161

6262

63-
@optools.add_to_registry()
63+
@optools.add_to_registry(aliases=['kd.make_namedtuple'])
6464
@optools.as_lambda_operator('kd.tuple.make_namedtuple')
6565
def make_namedtuple(**kwargs):
6666
"""Returns a namedtuple-like object containing the given `**kwargs`."""

0 commit comments

Comments
 (0)