split nek updates from PR 244 (#248)

* add nek files

* Fix order of imports.

* Fix spelling errors.

* Bunch of CI-demanded fixes.

* Formatting

---------

Co-authored-by: christian.lagemann <christian.lagemann@rwth-aachen.de>
Co-authored-by: Ludger Paehler <paehlerludger@gmail.com>

Author: 3 people

examples/nek/getting_started/3_pettingzoo/run_pettingzoo_docker.sh

@@ -1,6 +1,7 @@
 #!/usr/bin/env bash
 #
 # Run NEK5000 PettingZoo tests with MPMD coupling.
+# Config is loaded automatically from HuggingFace (environment_config.yaml).
 #
 # Usage:
 #     ./run_pettingzoo_docker.sh                    # Test only

examples/nek/getting_started/6_zeroshot_wing_demo/README.md

@@ -9,7 +9,6 @@ __This is a deployment/evaluation demo only (no training). The template and cont
 ## What the script does
 
 `test_nek_pettingzoo.py`:
-
 - loads a base `NekEnv` via `NekEnv.from_hf(...)` and wraps it with `make_pettingzoo_env(...)`
 - builds one controller per entry in `POLICY_SPECS` (from `meta_policy_small_wing_template.py`)
 - assigns each controller to actuator agents by `x_range` and `side` (`SS` means `y > 0`, `PS` means `y < 0`)
@@ -42,7 +41,6 @@ obs_dict, rewards_dict, terminations, truncations, infos = env.step(actions)
 ## Usage
 
 ### Recommended: use the runner script
-
 From `6_zeroshot_wing_demo/`:
 
 ```bash
@@ -58,7 +56,6 @@ mpirun -np 1 python test_nek_pettingzoo.py : -np 12 nek5000
 ```
 
 Legacy policy template + run root:
-
 ```bash
 mpirun -np 1 python test_nek_pettingzoo.py \
   --policy-template ./meta_policy_small_wing_template.py \
@@ -68,7 +65,6 @@ mpirun -np 1 python test_nek_pettingzoo.py \
 ```
 
 Useful overrides:
-
 - `--policy-template PATH` (defaults to `./meta_policy_small_wing_template.py`)
 - `--env ENV_NAME` (defaults from template `ENV_NAME`)
 - `--nproc NPROC` (defaults from template `NPROC`)
@@ -82,15 +78,13 @@ Useful overrides:
 The template defines a lightweight legacy-`MetaPolicy.py`-style configuration.
 
 Required top-level variables:
-
 - `ENV_NAME`
 - `NPROC`
 - `NUM_STEPS`
 - `POLICY_ROOT` (default for `--policy-root`)
 - `POLICY_SPECS` (list of policy group dicts)
 
 Each `POLICY_SPECS` entry supports:
-
 - `name`
 - `x_range: [x_min, x_max]`
 - `side: "SS"` (y>0) or `"PS"` (y<0)
@@ -101,7 +95,6 @@ Each `POLICY_SPECS` entry supports:
 - RL algorithms only: `agent_run_name`, `policy`, and/or `model_path`
 
 Algorithm semantics:
-
 - `ZERO` outputs an all-zero action (no model needed)
 - `BL` outputs a constant action equal to `action_max` (no model needed)
 - `PPO`/`TD3`/`DDPG` load a Stable-Baselines3 model from `model_path`/`POLICY_ROOT`

examples/nek/getting_started/6_zeroshot_wing_demo/meta_policy_small_wing_template.py

@@ -6,6 +6,7 @@
 import scipy.io as sio
 
 from hydrogym.core import CallbackBase
+from hydrogym.nek.nek_lib.nek_utils import show_end, show_title
 
 
 def integrate(