This works when running locally. However, if I bake the module containing the task transform() into an image and refer to it in a remote run, no mater what I do, I keep getting

flytekit/core/type_engine.py", line 1060, in to_literal
    raise TypeError("Ambiguous choice of variant for union type")
...
flytekit/core/promise.py", line 937, in create_and_link_node
    raise AssertionError(f"Failed to Bind variable {k} for function {entity.name}.") from e
AssertionError: Failed to Bind variable encoder for function dai_mle_models.onehotenc.simple_onehotenc.transform.