update code

Author: Jerry-Jzy

deepxde/data/pde_operator.py

@@ -268,19 +268,15 @@ def _losses(self, outputs, loss_fn, inputs, model, num_func, aux=None):
 
             def forward_call(trunk_input):
                 output = aux[0]((inputs[0], trunk_input))
-                return bkd.reshape(
-                    output, (batchsize_total, model.net.num_outputs)
-                )
+                return bkd.reshape(output, (batchsize_total, model.net.num_outputs))
 
             f = []
             if self.pde.pde is not None:
                 # Each f has the shape (N1, N2)
                 f = self.pde.pde(
                     inputs[1],
                     (
-                        bkd.reshape(
-                            outputs, (batchsize_total, model.net.num_outputs)
-                        ),
+                        bkd.reshape(outputs, (batchsize_total, model.net.num_outputs)),
                         forward_call,
                     ),
                     bkd.reshape(