diff --git a/arkindex/ponos/serializers.py b/arkindex/ponos/serializers.py index 7ce0b00f224203fe041c8f379fed541f20e6c13a..387a4c847951c3dfa6e42cfcdad567a557275b58 100644 --- a/arkindex/ponos/serializers.py +++ b/arkindex/ponos/serializers.py @@ -458,6 +458,7 @@ class TaskDefinitionSerializer(serializers.ModelSerializer): image_artifact_url = serializers.SerializerMethodField() s3_logs_put_url = serializers.SerializerMethodField() extra_files = serializers.DictField(default={}) + state = EnumField(State) @extend_schema_field(serializers.URLField(allow_null=True)) def get_image_artifact_url(self, task): @@ -497,6 +498,7 @@ class TaskDefinitionSerializer(serializers.ModelSerializer): "process_id", "gpu_id", "extra_files", + "state", ) read_only_fields = fields diff --git a/arkindex/ponos/tests/test_api.py b/arkindex/ponos/tests/test_api.py index 77102209c90182475db883ffca13205ddf090392..41fa597bee30476516d5ac7ee5de6a740abf1a30 100644 --- a/arkindex/ponos/tests/test_api.py +++ b/arkindex/ponos/tests/test_api.py @@ -351,6 +351,7 @@ class TestAPI(FixtureAPITestCase): 'process_id': str(self.process.id), 'extra_files': {}, 's3_logs_put_url': None, + 'state': State.Unscheduled.value, }) def test_task_definition_requires_login(self): @@ -409,6 +410,7 @@ class TestAPI(FixtureAPITestCase): 'process_id': str(self.process.id), 'extra_files': {}, 's3_logs_put_url': None, + 'state': State.Unscheduled.value, }) def test_task_definition_shm_size(self): @@ -444,6 +446,7 @@ class TestAPI(FixtureAPITestCase): 'process_id': str(self.process.id), 'extra_files': {}, 's3_logs_put_url': None, + 'state': State.Unscheduled.value, }) @patch("arkindex.project.aws.s3")