@@ -17,7 +17,7 @@ def test_model_run(client, configured_project_with_label, rand_gen):
1717 config = {"batch_size" : 100 , "reruns" : None }
1818 model_run = model .create_model_run (name , config )
1919 assert model_run .name == name
20- assert model_run .training_metadata ["batchSize " ] == config ["batch_size" ]
20+ assert model_run .training_metadata ["batch_size " ] == config ["batch_size" ]
2121 assert model_run .training_metadata ["reruns" ] == config ["reruns" ]
2222 assert model_run .model_id == model .uid
2323 assert model_run .created_by_id == client .get_user ().uid
@@ -59,7 +59,7 @@ def test_model_run_delete(client, model_run):
5959def test_model_run_update_config (model_run_with_training_metadata ):
6060 new_config = {"batch_size" : 2000 }
6161 res = model_run_with_training_metadata .update_config (new_config )
62- assert res ["trainingMetadata" ]["batchSize " ] == new_config ["batch_size" ]
62+ assert res ["trainingMetadata" ]["batch_size " ] == new_config ["batch_size" ]
6363
6464
6565def test_model_run_reset_config (model_run_with_training_metadata ):
@@ -71,7 +71,7 @@ def test_model_run_get_config(model_run_with_training_metadata):
7171 new_config = {"batch_size" : 2000 }
7272 model_run_with_training_metadata .update_config (new_config )
7373 res = model_run_with_training_metadata .get_config ()
74- assert res ["trainingMetadata" ]["batchSize " ] == new_config ["batch_size" ]
74+ assert res ["trainingMetadata" ]["batch_size " ] == new_config ["batch_size" ]
7575
7676
7777def test_model_run_data_rows_delete (client , model_run_with_model_run_data_rows ):
0 commit comments