Add type annotations

delta-io · Oct 11, 2023 · 531f3ed · 531f3ed
1 parent 6d458a7
commit 531f3ed
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/delta/tests/test_deltatable.py b/python/delta/tests/test_deltatable.py
@@ -19,7 +19,7 @@
 from typing import List, Set, Dict, Optional, Any, Callable, Union, Tuple
 from multiprocessing.pool import ThreadPool
 
-from pyspark.sql import DataFrame, Row
+from pyspark.sql import DataFrame, Row, SparkSession
 from pyspark.sql.column import _to_seq  # type: ignore[attr-defined]
 from pyspark.sql.functions import col, lit, expr, floor
 from pyspark.sql.types import StructType, StructField, StringType, IntegerType, LongType, DataType
@@ -323,7 +323,7 @@ def test_merge_with_inconsistent_sessions(self) -> None:
         target_path = os.path.join(self.tempFile, "target")
         spark = self.spark
 
-        def f(spark):
+        def f(spark: SparkSession) -> None:
             spark.range(20) \
                 .withColumn("x", col("id")) \
                 .withColumn("y", col("id")) \