styles(solver): minor style changes
@@ -603,6 +603,7 @@ class SolverSpherical(BaseSolver):
valid_generator = GeneratorSpherical(512, r_min, r_max, method='equally-spaced-noisy')
self.r_min, self.r_max = r_min, r_max
+ self.enforcer = enforcer
super(SolverSpherical, self).__init__(
diff_eqs=pde_system,
@@ -621,8 +622,6 @@ class SolverSpherical(BaseSolver):
batch_size=batch_size,
)
- self.enforcer = enforcer
-
def _auto_enforce(self, net, cond, *coordinates):
r"""Enforce condition on network with inputs. If self.enforcer is set, use it.
Otherwise, fill cond.enforce() with as many arguments as needed.
