diff --git a/python/nutpie/sample.py b/python/nutpie/sample.py index dc6dc21..908977b 100644 --- a/python/nutpie/sample.py +++ b/python/nutpie/sample.py @@ -305,15 +305,22 @@ def in_marimo_notebook() -> bool: def _mo_write_internal(cell_id, stream, value: object) -> None: """Write to marimo cell given cell_id and stream.""" + import marimo + + if marimo.__version__ < "0.19.0": + # The old CellOp API is identical to new CellNotificationUtils + from marimo._messaging.ops import CellOp as CellNotificationUtils + else: + from marimo._messaging.notification_utils import CellNotificationUtils + from marimo._messaging.cell_output import CellChannel - from marimo._messaging.ops import CellOp from marimo._messaging.tracebacks import write_traceback from marimo._output import formatting output = formatting.try_format(value) if output.traceback is not None: write_traceback(output.traceback) - CellOp.broadcast_output( + CellNotificationUtils.broadcast_output( channel=CellChannel.OUTPUT, mimetype=output.mimetype, data=output.data,