From 138e95ea71775dd0cd0953b1d637aa41fd3ca65b Mon Sep 17 00:00:00 2001 From: Ajay John Alex Date: Thu, 2 May 2024 22:33:28 +0100 Subject: [PATCH 1/3] Updating the interpolate methods mode argument as per function documentation --- ivy/data_classes/array/experimental/layers.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/ivy/data_classes/array/experimental/layers.py b/ivy/data_classes/array/experimental/layers.py index daee018c7e382..1a47ab3445a60 100644 --- a/ivy/data_classes/array/experimental/layers.py +++ b/ivy/data_classes/array/experimental/layers.py @@ -732,6 +732,10 @@ def interpolate( "nearest_exact", "tf_area", "bicubic", + "mitchellcubic", + "lanczos3", + "lanczos5", + "gaussian" ] ] = "linear", scale_factor: Optional[Union[Sequence[int], int]] = None, From 76b018ee4c3e8ccd1d122fb9613b04ffb2140e90 Mon Sep 17 00:00:00 2001 From: Ajay John Alex Date: Thu, 2 May 2024 22:55:01 +0100 Subject: [PATCH 2/3] Fixing #16396 for Union argument issue --- ivy/data_classes/array/experimental/layers.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/ivy/data_classes/array/experimental/layers.py b/ivy/data_classes/array/experimental/layers.py index 1a47ab3445a60..b8b7be4d57677 100644 --- a/ivy/data_classes/array/experimental/layers.py +++ b/ivy/data_classes/array/experimental/layers.py @@ -722,7 +722,7 @@ def interpolate( size: Union[Sequence[int], int], /, *, - mode: Union[ + mode: Literal[ "linear", "bilinear", @@ -736,8 +736,7 @@ def interpolate( "lanczos3", "lanczos5", "gaussian" - ] - ] = "linear", + ] = "linear", scale_factor: Optional[Union[Sequence[int], int]] = None, recompute_scale_factor: Optional[bool] = None, align_corners: Optional[bool] = None, From 13e75d715a926a73b8a3de7de7c086b5fea0dc38 Mon Sep 17 00:00:00 2001 From: ivy-branch Date: Thu, 2 May 2024 22:02:58 +0000 Subject: [PATCH 3/3] =?UTF-8?q?=F0=9F=A4=96=20Lint=20code?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- ivy/data_classes/array/experimental/layers.py | 29 +++++++++---------- 1 file changed, 14 insertions(+), 15 deletions(-) diff --git a/ivy/data_classes/array/experimental/layers.py b/ivy/data_classes/array/experimental/layers.py index b8b7be4d57677..ad9671eeccd15 100644 --- a/ivy/data_classes/array/experimental/layers.py +++ b/ivy/data_classes/array/experimental/layers.py @@ -722,21 +722,20 @@ def interpolate( size: Union[Sequence[int], int], /, *, - mode: - Literal[ - "linear", - "bilinear", - "trilinear", - "nearest", - "area", - "nearest_exact", - "tf_area", - "bicubic", - "mitchellcubic", - "lanczos3", - "lanczos5", - "gaussian" - ] = "linear", + mode: Literal[ + "linear", + "bilinear", + "trilinear", + "nearest", + "area", + "nearest_exact", + "tf_area", + "bicubic", + "mitchellcubic", + "lanczos3", + "lanczos5", + "gaussian", + ] = "linear", scale_factor: Optional[Union[Sequence[int], int]] = None, recompute_scale_factor: Optional[bool] = None, align_corners: Optional[bool] = None,