entity_category not applied to entity

[blakadder]

Editing device_customizes.py I've tried adding an entity_category to certain entities but that field is never applied. The entity applies other attributes like unit_of_measurement or device_class

configurations tried:

    'viomi.vacuum.v18:clean_use_time': {
        'value_ratio': 0.0166667,
        'device_class': 'duration',
        'entity_category': 'diagnostic',
},

    'viomi.vacuum.v18:repeat_state': {
        'entity_category': 'config',
    },

neither entity_category config works

[al-one]

Fixed in master branch.

[blakadder]

Still does not happen for devices in device_customisations.py. I only get the battery percentage in the diagnostic group despite setting multiple entities in custom

[al-one]

Fixed in 582fd22