improvement: add support for running in a marimo notebook

[akshayka]

This makes runhouse work in marimo notebooks, in addition to Jupyter notebooks.

[dongreenberg]

Love it! Thanks for the PR. One question - IIRC Marimo notebooks are .py files, so wouldn't they be importable natively on the server (and not require a short-circuit to write out the user's Module into a separate file)?

[akshayka]

Love it! Thanks for the PR. One question - IIRC Marimo notebooks are .py files, so wouldn't they be importable natively on the server (and not require a short-circuit to write out the user's Module into a separate file)?

They are Python files. But the function definitions are in their own lexical scopes (notebook files are lazy), so the server won't be able to find them — unless it knew it was looking at a marimo notebook, and knew how to extract the function. We'll have an API for that in the future but don't have it yet.

Maybe in the future we could have a way of defining top-level functions. But for now I think something like this PR is needed to get it to work.

[dongreenberg]

Ok! That makes sense, and appreciate the contribution to make Runhouse work with Marimo.