use is.numeric

Author: NarineK

R/pkg/R/DataFrame.R

@@ -602,8 +602,7 @@ setMethod("unpersist",
 setMethod("repartition",
           signature(x = "SparkDataFrame"),
           function(x, numPartitions = NULL, col = NULL, ...) {
-            if (!is.null(numPartitions) && (class(numPartitions) == "numeric" ||
-              class(numPartitions) == "integer")) {
+            if (!is.null(numPartitions) && is.numeric(numPartitions)) {
               # number of partitions and columns both are specified
               if (!is.null(col) && class(col) == "Column") {
                 cols <- list(col, ...)

R/pkg/R/RDD.R

@@ -1025,8 +1025,7 @@ setMethod("keyBy",
 setMethod("repartition",
           signature(x = "RDD"),
           function(x, numPartitions) {
-            if (!is.null(numPartitions) && (class(numPartitions) == "numeric" ||
-              class(numPartitions) == "integer")) {
+            if (!is.null(numPartitions) && is.numeric(numPartitions)) {
               coalesce(x, numPartitions, TRUE)
             } else {
               coalesce(x, 200L, TRUE)