diff --git a/ql/src/java/org/apache/hadoop/hive/ql/exec/spark/SparkUtilities.java b/ql/src/java/org/apache/hadoop/hive/ql/exec/spark/SparkUtilities.java index fac3cea..c69d07c 100644 --- a/ql/src/java/org/apache/hadoop/hive/ql/exec/spark/SparkUtilities.java +++ b/ql/src/java/org/apache/hadoop/hive/ql/exec/spark/SparkUtilities.java @@ -202,7 +202,7 @@ public static void collectOp(Collection> result, Operator root, C if (root == null) { return; } - if (clazz.equals(root.getClass())) { + if (clazz.equals(root.getClass()) || clazz.isAssignableFrom(root.getClass())) { result.add(root); } for (Operator child : root.getChildOperators()) {