Skip to content

Commit 13d04e6

Browse files
committed
Fix for ML transformer read/write
Signed-off-by: Jason T. Brown <[email protected]>
1 parent f550886 commit 13d04e6

File tree

1 file changed

+3
-3
lines changed
  • pyrasterframes/src/main/python/pyrasterframes

1 file changed

+3
-3
lines changed

pyrasterframes/src/main/python/pyrasterframes/rf_types.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@ class here provides the PyRasterFrames entry point.
3131

3232
from pyspark.ml.param.shared import HasInputCols
3333
from pyspark.ml.wrapper import JavaTransformer
34-
from pyspark.ml.util import JavaMLReadable, JavaMLWritable
34+
from pyspark.ml.util import JavaMLReadable, JavaMLWritable, DefaultParamsReadable, DefaultParamsWritable
3535

3636
from pyrasterframes.rf_context import RFContext
3737

@@ -462,7 +462,7 @@ def deserialize(self, datum):
462462
Tile.__UDT__ = TileUDT()
463463

464464

465-
class TileExploder(JavaTransformer, JavaMLReadable, JavaMLWritable):
465+
class TileExploder(JavaTransformer, DefaultParamsReadable, DefaultParamsWritable):
466466
"""
467467
Python wrapper for TileExploder.scala
468468
"""
@@ -472,7 +472,7 @@ def __init__(self):
472472
self._java_obj = self._new_java_obj("org.locationtech.rasterframes.ml.TileExploder", self.uid)
473473

474474

475-
class NoDataFilter(JavaTransformer, HasInputCols, JavaMLReadable, JavaMLWritable):
475+
class NoDataFilter(JavaTransformer, HasInputCols, DefaultParamsReadable, DefaultParamsWritable):
476476
"""
477477
Python wrapper for NoDataFilter.scala
478478
"""

0 commit comments

Comments
 (0)