From 4a0499ccf040ed018d4d6f4a641dc170963180d7 Mon Sep 17 00:00:00 2001
From: Tomas Pereira de Vasconcelos <tomasvasconcelos1@gmail.com>
Date: Fri, 2 Jul 2021 15:37:17 +0200
Subject: [PATCH] fix `pyspark.rdd.RDD.histogram`'s `buckets` argument (#549)

Closes #548
---
 third_party/3/pyspark/rdd.pyi | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/third_party/3/pyspark/rdd.pyi b/third_party/3/pyspark/rdd.pyi
index 3b43b74e..037975cd 100644
--- 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: ...