v2.0.0 container (ARM64) has a schema validation bug.
NVIDIA docs say to use:
“kind”: “GraphSAGE_XGBoost”
Source: Write your training configuration — Financial Fraud Training Container
Container defines the enum:
GRAPH_SAGE_XGB = “GraphSAGE_XGBoost” ✅
But excludes it from validation:
ModelConfig = Union[XGBSingleConfig, XGBListConfig, GNNBasedXGB] ❌
(Missing: GraphSAGEAndXGBConfig)
Result: Config validation fails with “Input should be ‘XGBoost’” or “Input should be ‘GNN_XGBoost’”
Fix needed:
ModelConfig = Union[XGBSingleConfig, XGBListConfig, GNNBasedXGB, GraphSAGEAndXGBConfig]
System: DGX Spark ARM64, v2.0.0 container