[Enh] Add numpy solver in config solvers (#869)

* support registering numpy config solvers when parsing attribute of numpy from config

* remove permutation code

* avoid duplicated register

Author: HydrogenSulfate

examples/NLS-MB/NLS-MB_optical_rogue_wave.py

@@ -417,20 +417,6 @@ def inference(cfg: DictConfig):
         store_key: output_dict[infer_key]
         for store_key, infer_key in zip(cfg.MODEL.output_keys, output_dict.keys())
     }
-    # TODO: Fix this mapping diff in dy2st
-    (
-        output_dict["Eu"],
-        output_dict["Ev"],
-        output_dict["eta"],
-        output_dict["pu"],
-        output_dict["pv"],
-    ) = (
-        output_dict["Eu"],
-        output_dict["Ev"],
-        output_dict["pu"],
-        output_dict["pv"],
-        output_dict["eta"],
-    )
 
     # visualize prediction
     Eu_true, Ev_true, pu_true, pv_true, eta_true = analytic_solution(input_dict)

examples/NLS-MB/NLS-MB_optical_soliton.py

@@ -394,20 +394,6 @@ def inference(cfg: DictConfig):
         store_key: output_dict[infer_key]
         for store_key, infer_key in zip(cfg.MODEL.output_keys, output_dict.keys())
     }
-    # TODO: Fix this mapping diff in dy2st
-    (
-        output_dict["Eu"],
-        output_dict["Ev"],
-        output_dict["eta"],
-        output_dict["pu"],
-        output_dict["pv"],
-    ) = (
-        output_dict["Eu"],
-        output_dict["Ev"],
-        output_dict["pu"],
-        output_dict["pv"],
-        output_dict["eta"],
-    )
 
     # visualize prediction
     Eu_true, Ev_true, pu_true, pv_true, eta_true = analytic_solution(input_dict)

ppsci/__init__.py

@@ -59,3 +59,16 @@
     "run_check_mesh",
     "lambdify",
 ]
+
+
+# NOTE: Register custom solvers for parsing values from omegaconf more flexible
+def _register_config_solvers():
+    import numpy as np
+    from omegaconf import OmegaConf
+
+    # register solver for "${numpy: xxx}" item, e.g. pi: "${numpy: pi}"
+    if not OmegaConf.has_resolver("numpy"):
+        OmegaConf.register_new_resolver("numpy", lambda x: getattr(np, x))
+
+
+_register_config_solvers()