File tree Expand file tree Collapse file tree 2 files changed +4
-8
lines changed Expand file tree Collapse file tree 2 files changed +4
-8
lines changed Original file line number Diff line number Diff line change @@ -83,7 +83,7 @@ class CreateAquaEvaluationDetails(Serializable):
83
83
ocpus : Optional [float ] = None
84
84
log_group_id : Optional [str ] = None
85
85
log_id : Optional [str ] = None
86
- metrics : Optional [List ] = None
86
+ metrics : Optional [List [ Dict [ str , Any ]] ] = None
87
87
force_overwrite : Optional [bool ] = False
88
88
89
89
class Config :
@@ -140,7 +140,7 @@ class AquaEvaluationCommands(Serializable):
140
140
evaluation_id : str
141
141
evaluation_target_id : str
142
142
input_data : Dict [str , Any ]
143
- metrics : List
143
+ metrics : List [ Dict [ str , Any ]]
144
144
output_dir : str
145
145
params : Dict [str , Any ]
146
146
Original file line number Diff line number Diff line change @@ -159,7 +159,8 @@ def create(
159
159
create_aqua_evaluation_details = CreateAquaEvaluationDetails (** kwargs )
160
160
except Exception as ex :
161
161
custom_errors = {
162
- "." .join (map (str , e ["loc" ])): e ["msg" ] for e in json .loads (ex .json ())
162
+ "." .join (map (str , e ["loc" ])): e ["msg" ]
163
+ for e in json .loads (ex .json ())
163
164
}
164
165
raise AquaValueError (
165
166
f"Invalid create evaluation parameters. Error details: { custom_errors } ."
@@ -619,11 +620,6 @@ def _build_launch_cmd(
619
620
evaluation_id = evaluation_id ,
620
621
evaluation_target_id = evaluation_source_id ,
621
622
input_data = {
622
- "columns" : {
623
- "prompt" : "prompt" ,
624
- "completion" : "completion" ,
625
- "category" : "category" ,
626
- },
627
623
"format" : Path (dataset_path ).suffix ,
628
624
"url" : dataset_path ,
629
625
},
You can’t perform that action at this time.
0 commit comments