Unverified Commit 39f6abe9 authored by Francisco Massa's avatar Francisco Massa Committed by GitHub
Browse files

Fix interpolate when no scale_factor is passed (#1785)

parent 5ac5ab9e
...@@ -77,7 +77,8 @@ def _check_size_scale_factor(dim, size, scale_factor): ...@@ -77,7 +77,8 @@ 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 not (scale_factor is not None and len(scale_factor) != dim): if scale_factor is not None and isinstance(scale_factor, tuple)\
and 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