gdal_calc python target?

databrickslabs · Nov 29, 2024 · 43618bf · 43618bf
1 parent ad11ba1
commit 43618bf
Show file tree

Hide file tree

Showing 6 changed files with 7 additions and 6 deletions.
diff --git a/python/mosaic/api/accessors.py b/python/mosaic/api/accessors.py
@@ -1,5 +1,5 @@
 from pyspark.sql import Column
-from pyspark.sql.functions import _to_java_column as pyspark_to_java_column
+from pyspark.sql.column import _to_java_column as pyspark_to_java_column
 
 from mosaic.config import config
 from mosaic.utils.types import ColumnOrName

diff --git a/python/mosaic/api/aggregators.py b/python/mosaic/api/aggregators.py
@@ -1,5 +1,5 @@
 from pyspark.sql import Column
-from pyspark.sql.functions import _to_java_column as pyspark_to_java_column
+from pyspark.sql.column import _to_java_column as pyspark_to_java_column
 
 from mosaic.config import config
 from mosaic.utils.types import ColumnOrName

diff --git a/python/mosaic/api/constructors.py b/python/mosaic/api/constructors.py
@@ -1,5 +1,5 @@
 from pyspark.sql import Column
-from pyspark.sql.functions import _to_java_column as pyspark_to_java_column
+from pyspark.sql.column import _to_java_column as pyspark_to_java_column
 
 from mosaic.config import config
 from mosaic.utils.types import ColumnOrName

diff --git a/python/mosaic/api/predicates.py b/python/mosaic/api/predicates.py
@@ -1,5 +1,5 @@
 from pyspark.sql import Column
-from pyspark.sql.functions import _to_java_column as pyspark_to_java_column
+from pyspark.sql.column import _to_java_column as pyspark_to_java_column
 
 from mosaic.config import config
 from mosaic.utils.types import ColumnOrName

diff --git a/python/mosaic/api/raster.py b/python/mosaic/api/raster.py
@@ -3,7 +3,7 @@
 from typing import Any
 
 from pyspark.sql import Column
-from pyspark.sql.functions import _to_java_column as pyspark_to_java_column
+from pyspark.sql.column import _to_java_column as pyspark_to_java_column
 from pyspark.sql.functions import lit
 
 from mosaic.config import config

diff --git a/python/test/test_mosaic.py b/python/test/test_mosaic.py
@@ -1,4 +1,5 @@
-from pyspark.sql.functions import _to_java_column, col
+from pyspark.sql.functions import col
+from pyspark.sql.column import _to_java_column
 
 from .context import MosaicContext, MosaicLibraryHandler
 from .utils import SparkTestCase