import explicitly

Author: wengh

tests/test_huggingface.py

@@ -1,6 +1,9 @@
 import pytest
 from pyspark.sql import SparkSession
 
+import pyspark_huggingface  # noqa: F401
+
+
 @pytest.fixture
 def spark():
     spark = SparkSession.builder.getOrCreate()

tests/test_huggingface_writer.py

@@ -6,6 +6,8 @@
 from pyspark.testing import assertDataFrameEqual
 from pytest_mock import MockerFixture
 
+import pyspark_huggingface  # noqa: F401
+
 # ============== Fixtures & Helpers ==============