formatting

Nike-Inc · Jun 4, 2024 · e61f37a · e61f37a
1 parent 969c901
commit e61f37a
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/koheesio/spark/utils.py b/src/koheesio/spark/utils.py
@@ -188,11 +188,15 @@ def import_pandas_based_on_pyspark_version():
     """
     try:
         import pandas as pd
+
         pyspark_version = get_spark_minor_version()
         pandas_version = pd.__version__
 
-        if (pyspark_version < 3.4 and pandas_version >= '2') or (pyspark_version >= 3.4 and pandas_version < '2'):
-            raise ImportError(f"For PySpark {pyspark_version}, please install Pandas version {'< 2' if pyspark_version < 3.4 else '>= 2'}")
+        if (pyspark_version < 3.4 and pandas_version >= "2") or (pyspark_version >= 3.4 and pandas_version < "2"):
+            raise ImportError(
+                f"For PySpark {pyspark_version}, "
+                f"please install Pandas version {'< 2' if pyspark_version < 3.4 else '>= 2'}"
+            )
 
         return pd
     except ImportError: