Support passing an operator to arolla.abc.set_expr_view_for_registered_operator(op)

apronchenkov · copybara-github · commit 584b9c462cc7 · 2026-01-27T21:55:51.000-08:00
PiperOrigin-RevId: 862039800
Change-Id: Iabc1ebfe3dc48a2057af64b5ab68a034bb665ff8
diff --git a/py/koladata/operators/arolla_bridge.py b/py/koladata/operators/arolla_bridge.py
@@ -28,12 +28,12 @@
 # NOTE: Implemented in C++ to allow bind-time literal evaluation.
 to_arolla_int64 = arolla.abc.lookup_operator('koda_internal.to_arolla_int64')
 arolla.abc.set_expr_view_for_registered_operator(
-    'koda_internal.to_arolla_int64', view.ArollaView
+    to_arolla_int64, view.ArollaView
 )
 
 to_arolla_text = arolla.abc.lookup_operator('koda_internal.to_arolla_text')
 arolla.abc.set_expr_view_for_registered_operator(
-    'koda_internal.to_arolla_text', view.ArollaView
+    to_arolla_text, view.ArollaView
 )
 
 

Original file line number	Diff line number	Diff line change
`@@ -28,12 +28,12 @@`
`28`	`28`	`# NOTE: Implemented in C++ to allow bind-time literal evaluation.`
`29`	`29`	`to_arolla_int64 = arolla.abc.lookup_operator('koda_internal.to_arolla_int64')`
`30`	`30`	`arolla.abc.set_expr_view_for_registered_operator(`
`31`		`- 'koda_internal.to_arolla_int64', view.ArollaView`
	`31`	`+ to_arolla_int64, view.ArollaView`
`32`	`32`	`)`
`33`	`33`
`34`	`34`	`to_arolla_text = arolla.abc.lookup_operator('koda_internal.to_arolla_text')`
`35`	`35`	`arolla.abc.set_expr_view_for_registered_operator(`
`36`		`- 'koda_internal.to_arolla_text', view.ArollaView`
	`36`	`+ to_arolla_text, view.ArollaView`
`37`	`37`	`)`
`38`	`38`
`39`	`39`