Skip to content

Commit 527687d

Browse files
Jack-Khuumalfet
authored andcommitted
Hide Eval specific args, unless eval is the subcommand (#885)
1 parent 583ae11 commit 527687d

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
@@ -230,6 +230,31 @@ def add_arguments_for_verb(parser, verb: str):
230230
choices=["fast", "cpu", "cuda", "mps"],
231231
help="Hardware device to use. Options: cpu, cuda, mps",
232232
)
233+
234+
if verb == "eval":
235+
_add_evaluation_args(parser)
236+
237+
parser.add_argument(
238+
"--hf-token",
239+
type=str,
240+
default=None,
241+
help="A HuggingFace API token to use when downloading model artifacts",
242+
)
243+
parser.add_argument(
244+
"--model-directory",
245+
type=Path,
246+
default=default_model_dir,
247+
help=f"The directory to store downloaded model artifacts. Default: {default_model_dir}",
248+
)
249+
parser.add_argument(
250+
"--port",
251+
type=int,
252+
default=5000,
253+
help="Port for the web server in browser mode",
254+
)
255+
256+
# Add CLI Args specific to Model Evaluation
257+
def _add_evaluation_args(parser) -> None:
233258
parser.add_argument(
234259
"--tasks",
235260
nargs="+",
@@ -249,24 +274,6 @@ def add_arguments_for_verb(parser, verb: str):
249274
default=None,
250275
help="Maximum length sequence to evaluate",
251276
)
252-
parser.add_argument(
253-
"--hf-token",
254-
type=str,
255-
default=None,
256-
help="A HuggingFace API token to use when downloading model artifacts",
257-
)
258-
parser.add_argument(
259-
"--model-directory",
260-
type=Path,
261-
default=default_model_dir,
262-
help=f"The directory to store downloaded model artifacts. Default: {default_model_dir}",
263-
)
264-
parser.add_argument(
265-
"--port",
266-
type=int,
267-
default=5000,
268-
help="Port for the web server in browser mode",
269-
)
270277

271278

272279
def arg_init(args):

0 commit comments

Comments
 (0)