Skip to content

Commit 89d82b1

Browse files
Jack-Khuufduwjj
authored andcommitted
Hide Eval specific args, unless eval is the subcommand (#885)
1 parent 552ab0f commit 89d82b1

File tree

1 file changed

+25
-18
lines changed

1 file changed

+25
-18
lines changed

cli.py

Lines changed: 25 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -236,6 +236,31 @@ def add_arguments_for_verb(parser, verb: str):
236236
choices=["fast", "cpu", "cuda", "mps"],
237237
help="Hardware device to use. Options: cpu, cuda, mps",
238238
)
239+
240+
if verb == "eval":
241+
_add_evaluation_args(parser)
242+
243+
parser.add_argument(
244+
"--hf-token",
245+
type=str,
246+
default=None,
247+
help="A HuggingFace API token to use when downloading model artifacts",
248+
)
249+
parser.add_argument(
250+
"--model-directory",
251+
type=Path,
252+
default=default_model_dir,
253+
help=f"The directory to store downloaded model artifacts. Default: {default_model_dir}",
254+
)
255+
parser.add_argument(
256+
"--port",
257+
type=int,
258+
default=5000,
259+
help="Port for the web server in browser mode",
260+
)
261+
262+
# Add CLI Args specific to Model Evaluation
263+
def _add_evaluation_args(parser) -> None:
239264
parser.add_argument(
240265
"--tasks",
241266
nargs="+",
@@ -255,24 +280,6 @@ def add_arguments_for_verb(parser, verb: str):
255280
default=None,
256281
help="Maximum length sequence to evaluate",
257282
)
258-
parser.add_argument(
259-
"--hf-token",
260-
type=str,
261-
default=None,
262-
help="A HuggingFace API token to use when downloading model artifacts",
263-
)
264-
parser.add_argument(
265-
"--model-directory",
266-
type=Path,
267-
default=default_model_dir,
268-
help=f"The directory to store downloaded model artifacts. Default: {default_model_dir}",
269-
)
270-
parser.add_argument(
271-
"--port",
272-
type=int,
273-
default=5000,
274-
help="Port for the web server in browser mode",
275-
)
276283

277284

278285
def arg_init(args):

0 commit comments

Comments
 (0)