Update comments for test cases

jingyu-ml · jingyu-ml · commit 5318241bb5ed · 2025-09-30T18:46:10.000Z
Signed-off-by: Jingyu Xin &lt;jingyux@nvidia.com&gt;
diff --git a/tests/gpu/torch/peft/test_megatron_peft.py b/tests/gpu/torch/peft/test_megatron_peft.py
@@ -237,6 +237,9 @@ def test_forward_with_two_loras(lora_config_1, lora_config_2):
     )
 
 
+# TODO: Save and restore with 1 or 2 GPUs
+# TODO: Grad check
+
 # def test_edge_cases_and_error_handling():
 #     """Test edge cases and error scenarios."""
 #     hidden_size = 320

Original file line number	Diff line number	Diff line change
`@@ -237,6 +237,9 @@ def test_forward_with_two_loras(lora_config_1, lora_config_2):`
`237`	`237`	`)`
`238`	`238`
`239`	`239`
	`240`	`+# TODO: Save and restore with 1 or 2 GPUs`
	`241`	`+# TODO: Grad check`
	`242`	`+`
`240`	`243`	`# def test_edge_cases_and_error_handling():`
`241`	`244`	`# """Test edge cases and error scenarios."""`
`242`	`245`	`# hidden_size = 320`