Yeelight - module 'yeelight' has no attribute 'enums'

Hi

First if all - I have updated my config per the latest changes to the yeelight component but they are not appearing when I turn discovery off and name my lights.

This is the error I get in my log. Any pointers?

I did not have this issue with 0.90 when the change was made (worked with the config change) so I have it setup correctly unless there is something else at play?

2019-05-01 07:01:19 ERROR (MainThread) [homeassistant.components.light] yeelight: Error on device update!
Traceback (most recent call last):
  File "/srv/homeassistant/lib/python3.6/site-packages/homeassistant/helpers/entity_platform.py", line 261, in _async_add_entity
    await entity.async_device_update(warning=False)                                                                                                                                     File "/srv/homeassistant/lib/python3.6/site-packages/homeassistant/helpers/entity.py", line 377, in async_device_update                                                                 await self.hass.async_add_executor_job(self.update)                                                                                                                                 File "/usr/local/lib/python3.6/asyncio/futures.py", line 331, in __iter__                                                                                                               yield self  # This tells Task to wait for completion.
  File "/usr/local/lib/python3.6/asyncio/tasks.py", line 244, in _wakeup                                                                                                                  future.result()
  File "/usr/local/lib/python3.6/asyncio/futures.py", line 244, in result                                                                                                                 raise self._exception
  File "/usr/local/lib/python3.6/concurrent/futures/thread.py", line 55, in run
    result = self.fn(*self.args, **self.kwargs)
  File "/srv/homeassistant/lib/python3.6/site-packages/homeassistant/components/yeelight/light.py", line 355, in update
    elif self.light_type == yeelight.enums.LightType.Ambient:
AttributeError: module 'yeelight' has no attribute 'enums'

In case anyone else suffers the same problem