diff --git a/nemo_gym/cli/env.py b/nemo_gym/cli/env.py index 5a6b12b8ab..038d8727b1 100644 --- a/nemo_gym/cli/env.py +++ b/nemo_gym/cli/env.py @@ -745,7 +745,7 @@ def init_resources_server(): # pragma: no cover jsonl_fpath: resources_servers/{server_type_name}/data/train.jsonl # local data file for this split num_repeats: 1 # times to repeat each example (e.g. for pass@k / mean@k) license: Apache 2.0 # required for train/validation; must be an allowed license string - # to fetch this split from a registry instead, add gitlab_identifier: or huggingface_identifier: + # to fetch this split from a registry instead, add a source: block (type: gitlab | huggingface) - name: validation type: validation jsonl_fpath: resources_servers/{server_type_name}/data/validation.jsonl diff --git a/tests/unit_tests/test_cli.py b/tests/unit_tests/test_cli.py index 654c847c8e..3df2feb43d 100644 --- a/tests/unit_tests/test_cli.py +++ b/tests/unit_tests/test_cli.py @@ -169,6 +169,12 @@ def test_init_resources_server_includes_domain(self) -> None: # This should not raise an assertion error about missing domain instance_config = ResourcesServerInstanceConfig.model_validate(full_config_dict) assert instance_config is not None + + # The generated config points users at the unified `source:` identifier, not the + # deprecated gitlab_identifier/huggingface_identifier. + assert "source:" in config_text + assert "gitlab_identifier" not in config_text + assert "huggingface_identifier" not in config_text finally: # Clean up the test server directory if server_path.exists():