Axon embedding layer compilation error

[zeio]

The problems was solved after I changed the format of data which I pass to the model from

{{batch.entities, batch.relations}, batch.true_labels}

to

{%{"entities" => batch.entities, "relations" => batch.relations}, batch.true_labels}