From 396270d50fbeaa9a99982e8c77fa9fadeb59e1de Mon Sep 17 00:00:00 2001 From: pkim1818 Date: Mon, 13 Nov 2023 20:36:11 -0800 Subject: [PATCH] fixing build call in proximal objective. no longer uses eq as a parameter, and replaces with use_jit. --- desc/optimize/_constraint_wrappers.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/desc/optimize/_constraint_wrappers.py b/desc/optimize/_constraint_wrappers.py index 518d262bb1..672a920f94 100644 --- a/desc/optimize/_constraint_wrappers.py +++ b/desc/optimize/_constraint_wrappers.py @@ -482,12 +482,12 @@ def build(self, eq=None, use_jit=None, verbose=1): # noqa: C901 # with this directly, so if the user wants to manually rebuild they should # do it before this wrapper is created for them. if not self._objective.built: - self._objective.build(self._eq, verbose=verbose) + self._objective.build(use_jit=use_jit, verbose=verbose) if not self._constraint.built: - self._constraint.build(self._eq, verbose=verbose) + self._constraint.build(use_jit=use_jit, verbose=verbose) for constraint in self._linear_constraints: - constraint.build(self._eq, verbose=verbose) + constraint.build(use_jit=use_jit, verbose=verbose) self._objectives = combine_args(self._objective, self._constraint) self._set_things(self._all_things)