Update pyspark_huggingface/compat/datasource.py

Co-authored-by: Haoyu Weng <wenghy02@gmail.com>

Author: lhoestq

pyspark_huggingface/compat/datasource.py

@@ -3,7 +3,7 @@
 import pyspark
 
 
-if int(pyspark.__version__.split(".")[0]) >= 4 and ("dev0" not in pyspark.__version__ and "dev1" not in pyspark.__version__):
+if int(pyspark.__version__.split(".")[0]) >= 4 and pyspark.__version__ not in ("4.0.0.dev0", "4.0.0.dev1"):
     from pyspark.sql.datasource import DataSource, DataSourceArrowWriter, DataSourceReader, DataSourceWriter, InputPartition, WriterCommitMessage
 else:
     class DataSource: