diff options
| -rwxr-xr-x | bin/chroma-cam | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/bin/chroma-cam b/bin/chroma-cam index 25f0eb3..c75f28f 100755 --- a/bin/chroma-cam +++ b/bin/chroma-cam @@ -32,10 +32,13 @@ if __name__ == '__main__': obj = mesh_from_stl(args[0]) else: module_name, function_name = args[0].rsplit('.', 1) - + orig_sys_path = list(sys.path) try: + sys.path.append('.') module = __import__(module_name, fromlist=[function_name]) + sys.path = orig_sys_path except ImportError: + sys.path = orig_sys_path raise obj = getattr(module, function_name) |