.

Author: ZiyiTsang

areal/api/cli_args.py

@@ -2066,7 +2066,7 @@ def to_structured_cfg(cfg, config_cls):
 
 def _validate_cfg(cfg: Any) -> None:
     """Validate configuration for consistency and correctness."""
-    
+    from areal.api.alloc_mode import AllocationMode
 
     am = AllocationMode.from_str(cfg.allocation_mode)
     if am.train_backend == "megatron" and cfg.actor.optimizer.type == "adam_bf16":

docs/cli_reference.md

@@ -739,14 +739,14 @@ Configuration for experiment statistics logging and tracking services.
 
 Configuration for SwanLab experiment tracking and monitoring.
 
-| Parameter | Type           | Default                   | Description |
-| --------- | -------------- | ------------------------- | ----------- |
-| `project` | string \| None | `None`                    | -           |
-| `name`    | string \| None | `None`                    | -           |
-| `config`  | `dict` \| None | `None`                    | -           |
-| `logdir`  | string \| None | `None`                    | -           |
-| `mode`    | string \| None | `"disabled"`              | -           |
-| `api_key` | string \| None | `None`                    | -           |
+| Parameter | Type           | Default      | Description |
+| --------- | -------------- | ------------ | ----------- |
+| `project` | string \| None | `None`       | -           |
+| `name`    | string \| None | `None`       | -           |
+| `config`  | `dict` \| None | `None`       | -           |
+| `logdir`  | string \| None | `None`       | -           |
+| `mode`    | string \| None | `"disabled"` | -           |
+| `api_key` | string \| None | `None`       | -           |
 
 (section-tensor-board)=