diff --git a/CHANGELOG.md b/CHANGELOG.md index 7db320d..ff58c08 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -10,6 +10,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0 ### Changed - This file and CONTRIBUTING.rst to markdown +- Updated inference status enum from numerical to meaningful string values ### Fixed diff --git a/datacrunch/InferenceClient/inference_client.py b/datacrunch/InferenceClient/inference_client.py index 216a057..df3fb05 100644 --- a/datacrunch/InferenceClient/inference_client.py +++ b/datacrunch/InferenceClient/inference_client.py @@ -12,11 +12,11 @@ class InferenceClientError(Exception): pass -class AsyncStatus(int, Enum): - Initialized = 0 - Queue = 1 - Inference = 2 - Completed = 3 +class AsyncStatus(str, Enum): + Initialized = "Initialized" + Queue = "Queue" + Inference = "Inference" + Completed = "Completed" @dataclass_json(undefined=Undefined.EXCLUDE)