Coverage for mindsdb / utilities / ml_task_queue / const.py: 100%

1from enum import Enum 

2 

3 

4TASKS_STREAM_NAME = b'ml-tasks' 

5TASKS_STREAM_CONSUMER_GROUP_NAME = 'ml_executors' 

6TASKS_STREAM_CONSUMER_NAME = 'ml_executor' 

7 

8 

9class ML_TASK_TYPE(Enum): 

10 LEARN = b'learn' 

11 PREDICT = b'predict' 

12 FINETUNE = b'finetune' 

13 DESCRIBE = b'describe' 

14 CREATE_VALIDATION = b'create_validation' 

15 CREATE_ENGINE = b'create_engine' 

16 UPDATE_ENGINE = b'update_engine' 

17 UPDATE = b'update' 

18 FUNC_CALL = b'func_call' 

19 

20 

21class ML_TASK_STATUS(Enum): 

22 WAITING = b'waiting' 

23 PROCESSING = b'processing' 

24 COMPLETE = b'complete' 

25 ERROR = b'error' 

26 TIMEOUT = b'timeout'