From 3b4caa0a058b7f3bf12f794a91023a7fb1476041 Mon Sep 17 00:00:00 2001 From: "Robert (Bobby) Evans" Date: Thu, 5 Aug 2021 07:53:56 -0500 Subject: [PATCH] Updated RunningWindow to support extended types too Signed-off-by: Robert (Bobby) Evans --- .../nvidia/spark/rapids/shims/spark301db/Spark301dbShims.scala | 3 ++- .../nvidia/spark/rapids/shims/spark311db/Spark311dbShims.scala | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/shims/spark301db/src/main/scala/com/nvidia/spark/rapids/shims/spark301db/Spark301dbShims.scala b/shims/spark301db/src/main/scala/com/nvidia/spark/rapids/shims/spark301db/Spark301dbShims.scala index 14f031b9f0a..730994e5931 100644 --- a/shims/spark301db/src/main/scala/com/nvidia/spark/rapids/shims/spark301db/Spark301dbShims.scala +++ b/shims/spark301db/src/main/scala/com/nvidia/spark/rapids/shims/spark301db/Spark301dbShims.scala @@ -121,7 +121,8 @@ class Spark301dbShims extends Spark301Shims { "i.e. (UNBOUNDED PRECEDING TO CURRENT ROW)", ExecChecks( (TypeSig.commonCudfTypes + TypeSig.NULL + TypeSig.DECIMAL + - TypeSig.STRUCT + TypeSig.ARRAY).nested() + + TypeSig.STRUCT + TypeSig.ARRAY + TypeSig.MAP).nested() + + TypeSig.psNote(TypeEnum.MAP, "Not supported as a partition by key") + TypeSig.psNote(TypeEnum.STRUCT, "Not supported as a partition by key") + TypeSig.psNote(TypeEnum.ARRAY, "Not supported as a partition by key"), TypeSig.all), diff --git a/shims/spark311db/src/main/scala/com/nvidia/spark/rapids/shims/spark311db/Spark311dbShims.scala b/shims/spark311db/src/main/scala/com/nvidia/spark/rapids/shims/spark311db/Spark311dbShims.scala index ff2ebbfaffd..6b86fa801ca 100644 --- a/shims/spark311db/src/main/scala/com/nvidia/spark/rapids/shims/spark311db/Spark311dbShims.scala +++ b/shims/spark311db/src/main/scala/com/nvidia/spark/rapids/shims/spark311db/Spark311dbShims.scala @@ -109,7 +109,8 @@ class Spark311dbShims extends Spark311Shims { "i.e. (UNBOUNDED PRECEDING TO CURRENT ROW)", ExecChecks( (TypeSig.commonCudfTypes + TypeSig.NULL + TypeSig.DECIMAL + - TypeSig.STRUCT + TypeSig.ARRAY).nested() + + TypeSig.STRUCT + TypeSig.ARRAY + TypeSig.MAP).nested() + + TypeSig.psNote(TypeEnum.MAP, "Not supported as a partition by key") + TypeSig.psNote(TypeEnum.STRUCT, "Not supported as a partition by key") + TypeSig.psNote(TypeEnum.ARRAY, "Not supported as a partition by key"), TypeSig.all),