update code

lululxvi · Dec 30, 2024 · bf69954 · bf69954
1 parent af99377
commit bf69954
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/deepxde/data/pde_operator.py b/deepxde/data/pde_operator.py
@@ -278,7 +278,7 @@ def forward_call(trunk_input):
                 output = aux[0]((inputs[0], trunk_input))
                 if not is_multi_outputs:
                     return output.reshape(shape0 * shape1, 1)
-                elif is_multi_outputs:
+                else:
                     return output.reshape(shape0 * shape1, shape2)
 
             if not is_multi_outputs:

diff --git a/deepxde/gradients/jacobian.py b/deepxde/gradients/jacobian.py
@@ -2,7 +2,6 @@
 
 from .. import config
 from ..backend import backend_name
-from .. import backend as bkd
 
 class Jacobian(ABC):
     """Compute `Jacobian matrix <https://en.wikipedia.org/wiki/Jacobian_matrix_and_determinant>`_