fix lint

joblibspark/utils.py

@@ -17,7 +17,7 @@
 """
 The utils functions for joblib spark backend.
 """
-
+from packaging.version import Version
 
 # pylint: disable=import-outside-toplevel
 def get_spark_session():
@@ -42,6 +42,9 @@ def create_resource_profile(num_cpus_worker_node, num_gpus_worker_node):
     :param num_gpus_worker_node: Number of gpus for the Spark worker node.
     :return: Spark ResourceProfile
     """
+
+    if Version(pyspark.__version__).release < (3, 1, 0):
+        return None
     from pyspark.resource.profile import ResourceProfileBuilder
     from pyspark.resource.requests import TaskResourceRequests