Bug: `st_geomfromwkb` no attribute `config.mosaic_context`

[aw-west-defra]

Describe the bug I cannot load my data from WKB in python. AttributeError: module 'mosaic.config.config' has no attribute 'mosaic_context'

To Reproduce

import mosaic
filepath = ...
df = (spark.read.parquet(filepath)
  .withColumn('geometry', st_geomfromwkb('geometry'))
)

Additional context The parquet file loads correctly as WKB. Briefly looking at mosaic.config.config, I think you use : instead of =.

Screenshots

Trackback

AttributeError                            Traceback (most recent call last)
<command-2659424542065347> in <module>
      2 f = 'dbfs:/tmp/osm_water.parquet'  # GeoParquet single file
      3 df = (spark.read.parquet(f)
----> 4   .withColumn('geometry', mosaic.st_geomfromwkb('geometry'))
      5 )
      6 display(df)

/local_disk0/.ephemeral_nfs/envs/pythonEnv-070f49e2-6a19-484d-9c62-f41e6c4ea63a/lib/python3.8/site-packages/mosaic/api/constructors.py in st_geomfromwkb(geom)
    126 
    127     """
--> 128     return config.mosaic_context.invoke_function(
    129         "st_geomfromwkb", pyspark_to_java_column(geom)
    130     )

AttributeError: module 'mosaic.config.config' has no attribute 'mosaic_context'

[aw-west-defra]

mosaic.__version__ == 0.3.9

[aw-west-defra]

The whole api is failing.

[aw-west-defra]

Ops, my bad.

mosaic.enable_mosaic(spark, dbutils)