AttributeError: module 'tensorflow_core.keras.layers' has no attribute 'Normalization'

[christine-petunia]

I work with Jupyter Lab. Tensorflow version: 2.0

AttributeError: module 'tensorflow_core.keras.layers' has no attribute 'Normalization'

[justinxzhao]

Hi Christine, which notebook are you referencing and which version of Ludwig are you using?

Ludwig has migrated to PyTorch, so there shouldn't be any Tensorflow errors. Perhaps some of our latest tutorials like this one would be helpful to get unblocked.

[connor-mccorm]

Hey @christine-petunia, just wanted to check in and see if you had any updates on this front. Was wondering if you could try updating to the latest version of Ludwig: 0.5.4 and let us know if this fixes the issue.