Skip to content

Commit 20cd628

Browse files
committed
Updated pr.
1 parent b983a7f commit 20cd628

File tree

2 files changed

+4
-22
lines changed

2 files changed

+4
-22
lines changed

ads/aqua/evaluation/entities.py

Lines changed: 4 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -95,24 +95,14 @@ class AquaEvalReport(Serializable):
9595
class Config:
9696
extra = "ignore"
9797

98-
class ModelParams(Serializable):
99-
max_tokens: str = ""
100-
top_p: str = ""
101-
top_k: str = ""
102-
temperature: str = ""
103-
presence_penalty: Optional[float] = 0.0
104-
frequency_penalty: Optional[float] = 0.0
105-
stop: Optional[Union[str, List[str]]] = Field(default_factory=list)
106-
model: Optional[str] = "odsc-llm"
107-
108-
class Config:
109-
extra = "allow"
110-
111-
class AquaEvalParams(ModelParams):
98+
class AquaEvalParams(Serializable):
11299
shape: str = ""
113100
dataset_path: str = ""
114101
report_path: str = ""
115102

103+
class Config:
104+
extra = "allow"
105+
116106
class AquaEvalMetric(Serializable):
117107
key: str
118108
name: str

tests/unitary/with_extras/aqua/test_evaluation.py

Lines changed: 0 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -506,17 +506,9 @@ def test_create_evaluation(
506506
"lifecycle_state": f"{evaluation_job_run.lifecycle_state}",
507507
"name": f"{evaluation_model.display_name}",
508508
"parameters": {
509-
"model": "odsc-llm",
510509
"dataset_path": "",
511-
"frequency_penalty": 0.0,
512-
"max_tokens": "",
513-
"presence_penalty": 0.0,
514510
"report_path": "",
515511
"shape": "",
516-
"stop": [],
517-
"temperature": "",
518-
"top_k": "",
519-
"top_p": "",
520512
},
521513
"source": {
522514
"id": "ocid1.datasciencemodel.oc1.iad.<OCID>",

0 commit comments

Comments
 (0)