Einops module not in setup.py

[asagilmore]

Hi All,

Just a heads up that the "einops" module is not listed in your setup.py, but running training without it installed raises a package not installed error. Installing manually fixes error.

Best,

Asa