ZCS: Add aux to _losses() due to base class upgrade (#1943)

lululxvi · Feb 23, 2025 · 179c1b9 · 179c1b9
1 parent 18400e5
commit 179c1b9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepxde/zcs/operator.py b/deepxde/zcs/operator.py
@@ -9,7 +9,7 @@
 class PDEOperatorCartesianProd(BasePDEOperatorCartesianProd):
     """Derived `PDEOperatorCartesianProd` class for ZCS support."""
 
-    def _losses(self, outputs, loss_fn, inputs, model, num_func):
+    def _losses(self, outputs, loss_fn, inputs, model, num_func, aux):
         # PDE
         f = []
         if self.pde.pde is not None: