File tree Expand file tree Collapse file tree 1 file changed +4
-19
lines changed Expand file tree Collapse file tree 1 file changed +4
-19
lines changed Original file line number Diff line number Diff line change @@ -867,18 +867,10 @@ def main(args: argparse.Namespace):
867
867
tokenizer_mode = tokenizer_mode ,
868
868
trust_remote_code = args .trust_remote_code )
869
869
870
- if args .dataset is not None :
871
- warnings .warn (
872
- "The '--dataset' argument will be deprecated in the next "
873
- "release. Please use '--dataset-name' and "
874
- "'--dataset-path' in the future runs." ,
875
- stacklevel = 2 )
876
- input_requests = sample_sharegpt_requests (
877
- dataset_path = args .dataset ,
878
- num_requests = args .num_prompts ,
879
- tokenizer = tokenizer ,
880
- fixed_output_len = args .sharegpt_output_len ,
881
- )
870
+ if args .dataset_name is None :
871
+ raise ValueError (
872
+ "Please specify '--dataset-name' and the corresponding "
873
+ "'--dataset-path' if required." )
882
874
883
875
elif args .dataset_name == "sharegpt" :
884
876
input_requests = sample_sharegpt_requests (
@@ -1052,13 +1044,6 @@ def main(args: argparse.Namespace):
1052
1044
default = "/v1/completions" ,
1053
1045
help = "API endpoint." ,
1054
1046
)
1055
- parser .add_argument (
1056
- "--dataset" ,
1057
- type = str ,
1058
- default = None ,
1059
- help = "Path to the ShareGPT dataset, will be deprecated in the "
1060
- "next release." ,
1061
- )
1062
1047
parser .add_argument (
1063
1048
"--dataset-name" ,
1064
1049
type = str ,
You can’t perform that action at this time.
0 commit comments