Update llm classifier guardrail experiment config paths

plstcharles-saifh · apalmas-saifh · commit 58c1f4eb73b3 · 2026-05-06T14:44:20.000-04:00
diff --git a/pyine/configs/experiment/guardrail/v0_classif_modernbert.yaml b/pyine/configs/experiment/guardrail/v0_classif_modernbert.yaml
@@ -4,9 +4,9 @@
 #
 # Usage:
 #   Single GPU:
-#     python -m pyine.apps.trainers.llm_classifier_trainer +experiment=guardrail/v0_modernbert
+#     python -m pyine.apps.trainers.llm_classifier_trainer +experiment=guardrail/v0_classif_modernbert
 #   Multi-GPU:
-#     torchrun --nproc_per_node=NUM_GPUS -m pyine.apps.trainers.llm_classifier_trainer +experiment=guardrail/v0_modernbert
+#     torchrun --nproc_per_node=NUM_GPUS -m pyine.apps.trainers.llm_classifier_trainer +experiment=guardrail/v0_classif_modernbert
 
 defaults:
   - /config/evals_config: correctness_taco_base
diff --git a/pyine/configs/experiment/guardrail/v0_classif_qwen2.yaml b/pyine/configs/experiment/guardrail/v0_classif_qwen2.yaml
@@ -2,16 +2,16 @@
 #
 # LLM classifier training: fine-tune Qwen2-0.5B as binary classifier
 #
-# Derives from v0_modernbert and overrides model + decoder-specific settings.
+# Derives from v0_classif_modernbert and overrides model + decoder-specific settings.
 #
 # Usage:
 #   Single GPU:
-#     python -m pyine.apps.trainers.llm_classifier_trainer +experiment=guardrail/v0_qwen2
+#     python -m pyine.apps.trainers.llm_classifier_trainer +experiment=guardrail/v0_classif_qwen2
 #   Multi-GPU:
-#     torchrun --nproc_per_node=NUM_GPUS -m pyine.apps.trainers.llm_classifier_trainer +experiment=guardrail/v0_qwen2
+#     torchrun --nproc_per_node=NUM_GPUS -m pyine.apps.trainers.llm_classifier_trainer +experiment=guardrail/v0_classif_qwen2
 
 defaults:
-  - guardrail/v0_modernbert
+  - guardrail/v0_classif_modernbert
   - _self_
 
 runtime:
diff --git a/tests/apps/trainers/test_llm_classifier_trainer_configs.py b/tests/apps/trainers/test_llm_classifier_trainer_configs.py
@@ -412,7 +412,7 @@ def test_modernbert_experiment_truncates_from_left(
         with hydra.initialize(config_path=None, version_base=pyine.configs.base.target_hydra_version):
             config = hydra.compose(
                 config_name="entrypoint",
-                overrides=["+experiment=guardrail/v0_modernbert"],
+                overrides=["+experiment=guardrail/v0_classif_modernbert"],
             )
 
         assert omegaconf.OmegaConf.select(config, "config.truncation_side") == "left"

Original file line number	Diff line number	Diff line change
`@@ -2,16 +2,16 @@`
`2`	`2`	`#`
`3`	`3`	`# LLM classifier training: fine-tune Qwen2-0.5B as binary classifier`
`4`	`4`	`#`
`5`		`-# Derives from v0_modernbert and overrides model + decoder-specific settings.`
	`5`	`+# Derives from v0_classif_modernbert and overrides model + decoder-specific settings.`
`6`	`6`	`#`
`7`	`7`	`# Usage:`
`8`	`8`	`# Single GPU:`
`9`		`-# python -m pyine.apps.trainers.llm_classifier_trainer +experiment=guardrail/v0_qwen2`
	`9`	`+# python -m pyine.apps.trainers.llm_classifier_trainer +experiment=guardrail/v0_classif_qwen2`
`10`	`10`	`# Multi-GPU:`
`11`		`-# torchrun --nproc_per_node=NUM_GPUS -m pyine.apps.trainers.llm_classifier_trainer +experiment=guardrail/v0_qwen2`
	`11`	`+# torchrun --nproc_per_node=NUM_GPUS -m pyine.apps.trainers.llm_classifier_trainer +experiment=guardrail/v0_classif_qwen2`
`12`	`12`
`13`	`13`	`defaults:`
`14`		`- - guardrail/v0_modernbert`
	`14`	`+ - guardrail/v0_classif_modernbert`
`15`	`15`	`- _self_`
`16`	`16`
`17`	`17`	`runtime:`
Original file line number	Diff line number	Diff line change
`@@ -412,7 +412,7 @@ def test_modernbert_experiment_truncates_from_left(`
`412`	`412`	`with hydra.initialize(config_path=None, version_base=pyine.configs.base.target_hydra_version):`
`413`	`413`	`config = hydra.compose(`
`414`	`414`	`config_name="entrypoint",`
`415`		`- overrides=["+experiment=guardrail/v0_modernbert"],`
	`415`	`+ overrides=["+experiment=guardrail/v0_classif_modernbert"],`
`416`	`416`	`)`
`417`	`417`
`418`	`418`	`assert omegaconf.OmegaConf.select(config, "config.truncation_side") == "left"`