Issue on start: No method get

[RubenKelevra]

I like to report an issue that seems to happen on startup of HA

Log message

Logger: homeassistant
Source: custom_components/ai_thermostat/climate.py:372
Integration: ai_thermostat (documentation, issues)
First occurred: 20:29:54 (1 occurrences)
Last logged: 20:29:54

Error doing job: Task exception was never retrieved
Traceback (most recent call last):
  File "/config/custom_components/ai_thermostat/climate.py", line 372, in _async_tvr_changed
    if self.hass.states.get(self.heater_entity_id).attributes.get('device').get('model') is not None:
AttributeError: 'NoneType' object has no attribute 'get'