Spaces:
Running
Running
| import os | |
| import sys | |
| import warnings | |
| # Check if JuliaCall is already loaded, and if so, warn the user | |
| # about the relevant environment variables. If not loaded, | |
| # set up sensible defaults. | |
| if "juliacall" in sys.modules: | |
| warnings.warn( | |
| "juliacall module already imported. " | |
| "Make sure that you have set the environment variable `PYTHON_JULIACALL_HANDLE_SIGNALS=yes` to avoid segfaults. " | |
| "Also note that PySR will not be able to configure `JULIA_NUM_THREADS` or `JULIA_OPTIMIZE` for you." | |
| ) | |
| else: | |
| # Required to avoid segfaults (https://juliapy.github.io/PythonCall.jl/dev/faq/) | |
| if os.environ.get("PYTHON_JULIACALL_HANDLE_SIGNALS", "yes") != "yes": | |
| warnings.warn( | |
| "PYTHON_JULIACALL_HANDLE_SIGNALS environment variable is set to something other than 'yes' or ''. " | |
| + "You will experience segfaults if running with multithreading." | |
| ) | |
| if os.environ.get("JULIA_NUM_THREADS", "auto") != "auto": | |
| warnings.warn( | |
| "JULIA_NUM_THREADS environment variable is set to something other than 'auto', " | |
| "so PySR was not able to set it. You may wish to set it to `'auto'` for full use " | |
| "of your CPU." | |
| ) | |
| # TODO: Remove these when juliapkg lets you specify this | |
| for k, default in ( | |
| ("PYTHON_JULIACALL_HANDLE_SIGNALS", "yes"), | |
| ("JULIA_NUM_THREADS", "auto"), | |
| ("JULIA_OPTIMIZE", "3"), | |
| ): | |
| os.environ[k] = os.environ.get(k, default) | |
| # Next, automatically load the juliacall extension if we're in a Jupyter notebook | |
| if os.environ.get("PYSR_AUTOLOAD_EXTENSIONS", "yes") in {"yes", ""}: | |
| try: | |
| get_ipython = sys.modules["IPython"].get_ipython | |
| if "IPKernelApp" not in get_ipython().config: | |
| raise ImportError("console") | |
| print( | |
| "Detected Jupyter notebook. Loading juliacall extension. Set `PYSR_AUTOLOAD_EXTENSIONS=no` to disable." | |
| ) | |
| # TODO: Turn this off if juliacall does this automatically | |
| get_ipython().run_line_magic("load_ext", "juliacall") | |
| except Exception: | |
| pass | |
| elif os.environ["PYSR_AUTOLOAD_EXTENSIONS"] not in {"no", "yes", ""}: | |
| warnings.warn( | |
| "PYSR_AUTOLOAD_EXTENSIONS environment variable is set to something other than 'yes' or 'no' or ''." | |
| ) | |
| from juliacall import Main as jl # type: ignore | |
| # Finally, overwrite the seval function to use Meta.parseall | |
| # instead of Meta.parse. | |
| if jl.seval('VERSION >= v"1.9.0-DEV.0"'): | |
| # TODO: Overwrite this once PythonCall.jl is updated: | |
| def seval(s: str): | |
| return jl.eval(jl.Meta.parseall(s)) | |
| jl.seval = seval | |
| jl.seval("using SymbolicRegression") | |
| SymbolicRegression = jl.SymbolicRegression | |