"tests/vscode:/vscode.git/clone" did not exist on "916d375ba3d62e018231633ca74e33ce128085c3"
Unverified Commit 2875315d authored by eellison's avatar eellison Committed by GitHub
Browse files

fix len error (#1981)


Co-authored-by: default avatareellison <eellison@fb.com>
parent 7c077f6a
...@@ -77,8 +77,9 @@ def _check_size_scale_factor(dim, size, scale_factor): ...@@ -77,8 +77,9 @@ def _check_size_scale_factor(dim, size, scale_factor):
raise ValueError("either size or scale_factor should be defined") raise ValueError("either size or scale_factor should be defined")
if size is not None and scale_factor is not None: if size is not None and scale_factor is not None:
raise ValueError("only one of size or scale_factor should be defined") raise ValueError("only one of size or scale_factor should be defined")
if scale_factor is not None and isinstance(scale_factor, tuple)\ if scale_factor is not None:
and len(scale_factor) != dim: if isinstance(scale_factor, (list, tuple)):
if len(scale_factor) != dim:
raise ValueError( raise ValueError(
"scale_factor shape must match input shape. " "scale_factor shape must match input shape. "
"Input is {}D, scale_factor size is {}".format(dim, len(scale_factor)) "Input is {}D, scale_factor size is {}".format(dim, len(scale_factor))
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment