From 1839ef9310e261a6f350d9aeedc0b5d053508d6a Mon Sep 17 00:00:00 2001 From: Amir Miron Date: Wed, 25 Feb 2015 10:34:04 +0200 Subject: [PATCH] fixed failed function teleportation when matching module did not exist on the other side --- rpyc/utils/teleportation.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/rpyc/utils/teleportation.py b/rpyc/utils/teleportation.py index 9c675b6a..fcb24817 100644 --- a/rpyc/utils/teleportation.py +++ b/rpyc/utils/teleportation.py @@ -117,7 +117,10 @@ def _import_codetup(codetup): def import_function(functup): name, modname, defaults, codetup = functup - mod = __import__(modname, None, None, "*") + try: + mod = __import__(modname, None, None, "*") + except ImportError: + mod = __import__("__main__", None, None, "*") codeobj = _import_codetup(codetup) return FunctionType(codeobj, mod.__dict__, name, defaults)