Remove duplicate and unused WorkerState declaration

Author: mattt

Diff for: python/cog/predictor.py

@@ -20,13 +20,14 @@
     cast,
     get_type_hints,
 )
-from unittest.mock import patch
 
 try:
     from typing import get_args, get_origin
 except ImportError:  # Python < 3.8
     from typing_compat import get_args, get_origin  # type: ignore
 
+from unittest.mock import patch
+
 import structlog
 import yaml
 from pydantic import BaseModel, Field, create_model

Diff for: python/cog/server/worker.py

@@ -8,7 +8,6 @@
 import types
 from collections import defaultdict
 from contextvars import ContextVar
-from enum import Enum, auto, unique
 from multiprocessing.connection import Connection
 from typing import Any, AsyncIterator, Callable, Iterator, Optional, TextIO
 
@@ -42,16 +41,6 @@
 _spawn = multiprocessing.get_context("spawn")
 
 
-@unique
-class WorkerState(Enum):
-    NEW = auto()
-    STARTING = auto()
-    IDLE = auto()
-    PROCESSING = auto()
-    BUSY = auto()
-    DEFUNCT = auto()
-
-
 class Mux:
     def __init__(self, terminating: asyncio.Event) -> None:
         self.outs: "defaultdict[str, asyncio.Queue[PublicEventType]]" = defaultdict(