[Fix] Fix dist_wrapper in solver (#978)

* do not wrap equation expression in distributed env

* update requirements.txt

Author: HydrogenSulfate

ppsci/solver/solver.py

@@ -517,10 +517,11 @@ def convert_expr(
                     for name in container.output_expr:
                         if isinstance(container.output_expr[name], sp.Basic):
                             container.output_expr[name] = funcs[ind]
-                            if self.world_size > 1:
-                                container.output_expr[name] = dist_wrapper(
-                                    container.output_expr[name]
-                                )
+                            # FIXME: Equation with parameter not support yet.
+                            # if self.world_size > 1:
+                            #     container.output_expr[name] = dist_wrapper(
+                            #         container.output_expr[name]
+                            #     )
                             ind += 1
 
         if self.constraint:

requirements.txt

@@ -5,7 +5,7 @@ hydra-core
 imageio
 matplotlib
 meshio==5.3.4
-numpy>=1.20.0,<=1.23.1
+numpy>=1.20.0,<2.0.0
 pydantic>=2.5.0
 pyevtk
 pyyaml