Call mo.output.replace from rust

[williambdean]

I'm trying to call mo.output functions from within rust

I'm seeing that the get_context function is not being detected

marimo/marimo/_runtime/context/types.py

Lines 226 to 234 in 77e9556

	def get_context() -> RuntimeContext:
	"""Return the runtime context.
	Throws a ContextNotInitializedError if the context has not been
	created.
	"""
	if _THREAD_LOCAL_CONTEXT.runtime_context is None:
	raise ContextNotInitializedError
	return _THREAD_LOCAL_CONTEXT.runtime_context

Any idea on a work around? Maybe I get get the current context from within python and create partial function

PR I'm working on is here: pymc-devs/nutpie#224

EDIT: I was able to get it working by replacing all get_context calls and passing ctx.stream directly. Required rewriting the logic of mo.output.replace

[dmadisetti]

Nice! Looked at your PR. A little tricky with callbacks. Using internals may be subject to change- so let us know if this breaks at any point

[williambdean]

Will do! thanks for taking a look

[williambdean]

Getting some issues with this now

pymc-devs/pymc#8162

Still understanding but think this was change from 0.18.0

[dmadisetti]

We had a pretty large restructure at this point. Feel free to tag me in any PRs.

---

An aside, but I hacked this together over a weekend awhile back: https://github.com/dmadisetti/anywidget-rs
Which compiles rust -> wasm and exposes it as an anywidget. You could use this (create a widget output and dynamically update it for outputs)- opposed to using output.append