Skip to content

Commit 7bc0b35

Browse files
update dict
1 parent 67586de commit 7bc0b35

File tree

2 files changed

+3
-4
lines changed

2 files changed

+3
-4
lines changed

ads/aqua/finetuning/finetuning.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -6,8 +6,7 @@
66
import json
77
import os
88
from dataclasses import asdict, fields
9-
from typing import Dict, List
10-
from collections import defaultdict
9+
from typing import Dict
1110

1211
from oci.data_science.models import (
1312
Metadata,
@@ -575,7 +574,7 @@ def get_finetuning_default_params(self, model_id: str) -> Dict:
575574
Dict of parameters from the loaded from finetuning config json file. If config information is not available,
576575
then an empty dict is returned.
577576
"""
578-
default_params = defaultdict(dict)
577+
default_params = {"params": {}}
579578
finetuning_config = self.get_finetuning_config(model_id)
580579
config_parameters = finetuning_config.get("configuration", UNKNOWN_DICT)
581580
dataclass_fields = {field.name for field in fields(AquaFineTuningParams)}

tests/unitary/with_extras/aqua/test_finetuning.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -262,7 +262,7 @@ def test_get_finetuning_default_params(self):
262262
# check when config json is not available
263263
self.app.get_finetuning_config = MagicMock(return_value={})
264264
result = self.app.get_finetuning_default_params(model_id="test_model_id")
265-
assert result == {}
265+
assert result == {"params": {}}
266266

267267
@parameterized.expand(
268268
[

0 commit comments

Comments
 (0)