add import

NVIDIA · Nov 17, 2021 · c8ac88b · c8ac88b
1 parent 7c0ef58
commit c8ac88b
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/integration_tests/src/main/python/string_test.py b/integration_tests/src/main/python/string_test.py
@@ -20,6 +20,7 @@
 from marks import *
 from pyspark.sql.types import *
 import pyspark.sql.functions as f
+from spark_session import is_before_spark_320
 
 def mk_str_gen(pattern):
     return StringGen(pattern).with_special_case('').with_special_pattern('.{0,10}')