Browse Source

!801 Fix cli for unexpected condition

Merge pull request !801 from moran/network_validation
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
dd636b76e4
1 changed files with 6 additions and 1 deletions
  1. +6
    -1
      mindinsight/mindconverter/cli.py

+ 6
- 1
mindinsight/mindconverter/cli.py View File

@@ -85,6 +85,11 @@ class OutputDirAction(argparse.Action):
option_string (str): Optional string for specific argument name. Default: None.
"""
output = values

if len(output) > ARGUMENT_LENGTH_LIMIT:
parser_in.error(
f"The length of {option_string}{output} should be no more than {ARGUMENT_LENGTH_LIMIT}.")

if output.startswith('~'):
output = os.path.realpath(os.path.expanduser(output))

@@ -204,7 +209,7 @@ class ShapeAction(argparse.Action):
in_shape = None
shape_str = values

shape_list = shape_str.split(';')
shape_list = shape_str.split(':')
if not len(shape_list) == EXPECTED_SHAPE_NUMBER:
parser_in.error(f"Only support one shape now, but get {len(shape_list)}.")



Loading…
Cancel
Save