diff --git a/py_clob_client/clob_types.py b/py_clob_client/clob_types.py index 8811a72..ce0877f 100644 --- a/py_clob_client/clob_types.py +++ b/py_clob_client/clob_types.py @@ -1,3 +1,4 @@ +from enum import Enum from typing import Any from dataclasses import dataclass, asdict from json import dumps @@ -9,7 +10,7 @@ from .constants import ZERO_ADDRESS -class OrderType(enumerate): +class OrderType(str, Enum): GTC = "GTC" FOK = "FOK" GTD = "GTD" @@ -184,7 +185,7 @@ def json(self): return dumps(self.__dict__, separators=(",", ":")) -class AssetType(enumerate): +class AssetType(str, Enum): COLLATERAL = "COLLATERAL" CONDITIONAL = "CONDITIONAL"