From 375d87ed8fd5161c8052ad8898be050fe150e095 Mon Sep 17 00:00:00 2001 From: Rory Conlin Date: Tue, 9 Jul 2024 20:31:42 -0400 Subject: [PATCH] Add check for invalid kwarg, avoid reusing common argname --- desc/compute/utils.py | 1 + desc/magnetic_fields/_core.py | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/desc/compute/utils.py b/desc/compute/utils.py index 346b2ccc57..491e4ebddf 100644 --- a/desc/compute/utils.py +++ b/desc/compute/utils.py @@ -127,6 +127,7 @@ def _compute( cannot give the argument basis='xyz' since that will break the recursion. In that case, either call above function or manually convert the output to xyz basis. """ + assert kwargs.get("basis", "rpz") == "rpz", "_compute only works in rpz coordinates" parameterization = _parse_parameterization(parameterization) if isinstance(names, str): names = [names] diff --git a/desc/magnetic_fields/_core.py b/desc/magnetic_fields/_core.py index 5c8808ef43..8cf4a4b35c 100644 --- a/desc/magnetic_fields/_core.py +++ b/desc/magnetic_fields/_core.py @@ -1841,7 +1841,7 @@ def compute( names, obj=self, grid=grid, - method=kwargs.pop("method", "auto"), + jitable=kwargs.pop("jitable", False), **kwargs, ) if data is None: