fix: update test to account for changes

Author: benieric

tests/unit/sagemaker/modules/train/test_model_trainer.py

@@ -504,19 +504,19 @@ def test_train_with_distributed_config(
         assert os.path.exists(expected_runner_json_path)
         with open(expected_runner_json_path, "r") as f:
             runner_json_content = f.read()
-            assert test_case["distributed"].model_dump(exclude_none=True) == (
+            assert test_case["distributed"].model_dump() == (
                 json.loads(runner_json_content)
             )
         assert os.path.exists(expected_source_code_json_path)
         with open(expected_source_code_json_path, "r") as f:
             source_code_json_content = f.read()
-            assert test_case["source_code"].model_dump(exclude_none=True) == (
+            assert test_case["source_code"].model_dump() == (
                 json.loads(source_code_json_content)
             )
         assert os.path.exists(expected_source_code_json_path)
         with open(expected_source_code_json_path, "r") as f:
             source_code_json_content = f.read()
-            assert test_case["source_code"].model_dump(exclude_none=True) == (
+            assert test_case["source_code"].model_dump() == (
                 json.loads(source_code_json_content)
             )
     finally: