fix: change dataset for sft trainer nova test

Syed Jafri · Syed Jafri · commit 2bbe72a5cbf4 · 2026-05-07T21:54:01.000Z
diff --git a/sagemaker-train/tests/integ/train/test_sft_trainer_integration.py b/sagemaker-train/tests/integ/train/test_sft_trainer_integration.py
@@ -99,7 +99,7 @@ def test_sft_trainer_nova_workflow(sagemaker_session_us_east_1):
         model_package_group="sdk-test-finetuned-models",
         mlflow_experiment_name="test-nova-finetuned-models-exp",
         mlflow_run_name="test-nova-finetuned-models-run",
-        training_dataset="s3://mc-flows-sdk-testing-us-east-1/input_data/sft-nova/sft_8_samples.jsonl",
+        training_dataset="s3://mc-flows-sdk-testing-us-east-1/input_data/sft-nova/sft_200_samples.jsonl",
         s3_output_path="s3://mc-flows-sdk-testing-us-east-1/output/",
         sagemaker_session=sagemaker_session_us_east_1
     )

Original file line number	Diff line number	Diff line change
`@@ -99,7 +99,7 @@ def test_sft_trainer_nova_workflow(sagemaker_session_us_east_1):`
`99`	`99`	`model_package_group="sdk-test-finetuned-models",`
`100`	`100`	`mlflow_experiment_name="test-nova-finetuned-models-exp",`
`101`	`101`	`mlflow_run_name="test-nova-finetuned-models-run",`
`102`		`- training_dataset="s3://mc-flows-sdk-testing-us-east-1/input_data/sft-nova/sft_8_samples.jsonl",`
	`102`	`+ training_dataset="s3://mc-flows-sdk-testing-us-east-1/input_data/sft-nova/sft_200_samples.jsonl",`
`103`	`103`	`s3_output_path="s3://mc-flows-sdk-testing-us-east-1/output/",`
`104`	`104`	`sagemaker_session=sagemaker_session_us_east_1`
`105`	`105`	`)`