Skip to content

Commit bc939f9

Browse files
committed
Fixes unit tests
1 parent bde6c10 commit bc939f9

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

tests/unitary/with_extras/aqua/test_deployment.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -486,6 +486,7 @@ class TestDataset:
486486
"models": [
487487
{
488488
"env_var": {},
489+
"params": {},
489490
"gpu_count": 2,
490491
"model_id": "test_model_id_1",
491492
"model_name": "test_model_1",
@@ -495,6 +496,7 @@ class TestDataset:
495496
},
496497
{
497498
"env_var": {},
499+
"params": {},
498500
"gpu_count": 2,
499501
"model_id": "test_model_id_2",
500502
"model_name": "test_model_2",
@@ -504,6 +506,7 @@ class TestDataset:
504506
},
505507
{
506508
"env_var": {},
509+
"params": {},
507510
"gpu_count": 2,
508511
"model_id": "test_model_id_3",
509512
"model_name": "test_model_3",
@@ -985,6 +988,7 @@ class TestDataset:
985988
multi_model_deployment_model_attributes = [
986989
{
987990
"env_var": {"--test_key_one": "test_value_one"},
991+
"params": {},
988992
"gpu_count": 1,
989993
"model_id": "ocid1.compartment.oc1..<OCID>",
990994
"model_name": "model_one",
@@ -994,6 +998,7 @@ class TestDataset:
994998
},
995999
{
9961000
"env_var": {"--test_key_two": "test_value_two"},
1001+
"params": {},
9971002
"gpu_count": 1,
9981003
"model_id": "ocid1.compartment.oc1..<OCID>",
9991004
"model_name": "model_two",
@@ -1003,6 +1008,7 @@ class TestDataset:
10031008
},
10041009
{
10051010
"env_var": {"--test_key_three": "test_value_three"},
1011+
"params": {},
10061012
"gpu_count": 1,
10071013
"model_id": "ocid1.compartment.oc1..<OCID>",
10081014
"model_name": "model_three",

0 commit comments

Comments
 (0)