Air Purifier zhimi.airpurifier.m1 set_mode isn't working

[krotoniec]

Hello,
I was playing with miio library and noticed that set_mode doesn't work:
from miio import*
pur = AirPurifier(ip='192.168.2.13', token="6106c6f3fa4357a1eacacc324685ac9a")
pur.set_mode(OperationMode.Silent)
Traceback (most recent call last):
File "", line 1, in
NameError: name 'OperationMode' is not defined

Am I missing something?

[syssi]

You should import the OperationMode class of the air purifier:

$ grep "class OperationMode" miio/*.py
miio/airconditioningcompanion.py:class OperationMode(enum.Enum):                                                                                                                                                                               
miio/airfresh.py:class OperationMode(enum.Enum):                                                                                                                                                                                               
miio/airhumidifier.py:class OperationMode(enum.Enum):                                                                                                                                                                                          
miio/airpurifier.py:class OperationMode(enum.Enum):                                                                                                                                                                                            
miio/cooker.py:class OperationMode(enum.Enum): 

[krotoniec]

Worked like a charm, Thanks!
from miio import*
from miio.airpurifier import AirPurifier, OperationMode
pur = AirPurifier(ip='192.168.2.120', token='1e0e11653f345434235345438e5435')
pur.set_mode(OperationMode.Auto)
['ok']