Merge pull request #521 from zero323/fail-on-stopiteration

Annotate fail_on_stopiteration

third_party/3/pyspark/_typing.pyi

@@ -16,9 +16,10 @@
 # specific language governing permissions and limitations
 # under the License.
 
-from typing import Iterable, Sized, TypeVar, Union
+from typing import Callable, Iterable, Sized, TypeVar, Union
 from typing_extensions import Protocol
 
+F = TypeVar("F", bound=Callable)
 T = TypeVar("T", covariant=True)
 
 PrimitiveType = Union[bool, float, int, str]

third_party/3/pyspark/util.pyi

@@ -15,21 +15,20 @@
 # KIND, either express or implied. See the License for the
 # specific language governing permissions and limitations
 # under the License.
-
-# Stubs for pyspark.util (Python 3.7)
 #
-# NOTE: This dynamically typed stub was automatically generated by stubgen.
+
+from typing import Any, Callable
+from pyspark._typing import F
 
 import threading
-from typing import Any
 
 def print_exec(stream: Any) -> None: ...
 
 class VersionUtils:
  @staticmethod
  def majorMinorVersion(sparkVersion: Any): ...
 
-def fail_on_stopiteration(f: Any): ...
+def fail_on_stopiteration(f: F) -> F: ...
 
 class InheritableThread(threading.Thread):
  def __init__(self, target: Any, *args: Any, **kwargs: Any): ...