diff --git a/.github/actions/python_build/action.yml b/.github/actions/python_build/action.yml
index 3eab1778d..d3e32635c 100644
--- a/.github/actions/python_build/action.yml
+++ b/.github/actions/python_build/action.yml
@@ -12,6 +12,7 @@ runs:
       run: |
         cd python
         pip install build wheel pyspark==${{ matrix.spark }} numpy==${{ matrix.numpy }}
+        pip install gdal==${{ matrix.gdal }}
         pip install .
     - name: Test and build python package
       shell: bash
diff --git a/python/setup.cfg b/python/setup.cfg
index 2bc75b1be..1d34b6c9a 100644
--- a/python/setup.cfg
+++ b/python/setup.cfg
@@ -24,7 +24,6 @@ setup_requires =
 install_requires =
     keplergl==0.3.2
     h3==3.7.3
-    gdal[numpy]==3.4.3
 
 [options.package_data]
 mosaic =
diff --git a/python/test/test_raster_functions.py b/python/test/test_raster_functions.py
index 077bd06df..910e5f5a1 100644
--- a/python/test/test_raster_functions.py
+++ b/python/test/test_raster_functions.py
@@ -1,7 +1,3 @@
-import logging
-import random
-import unittest
-
 from pyspark.sql.functions import abs, col, first, lit, sqrt, array
 
 from .context import api