fix pyspark.rdd.RDD.histogram's buckets argument (#549)

Closes #548
zero323 · Jul 2, 2021 · 4a0499c · 4a0499c
1 parent 33cab26
commit 4a0499c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/third_party/3/pyspark/rdd.pyi b/third_party/3/pyspark/rdd.pyi
@@ -218,7 +218,7 @@ class RDD(Generic[T]):
     def sum(self: RDD[NumberOrArray]) -> NumberOrArray: ...
     def count(self) -> int: ...
     def stats(self: RDD[NumberOrArray]) -> StatCounter: ...
-    def histogram(self, buckets: List[T]) -> Tuple[List[T], List[int]]: ...
+    def histogram(self, buckets: Union[int, List[T], Tuple[T, ...]]) -> Tuple[List[T], List[int]]: ...
     def mean(self: RDD[NumberOrArray]) -> NumberOrArray: ...
     def variance(self: RDD[NumberOrArray]) -> NumberOrArray: ...
     def stdev(self: RDD[NumberOrArray]) -> NumberOrArray: ...