Skip to content

Commit d9615f5

Browse files
committed
Change task type names to lower case
1 parent 1cac147 commit d9615f5

File tree

7 files changed

+19
-18
lines changed

7 files changed

+19
-18
lines changed

packages/examples/cvat/exchange-oracle/src/core/config.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -234,13 +234,13 @@ class CoreConfig:
234234
default_assignment_time = int(os.environ.get("DEFAULT_ASSIGNMENT_TIME", 1800))
235235

236236
skeleton_assignment_size_mult = int(os.environ.get("SKELETON_ASSIGNMENT_SIZE_MULT", 1))
237-
"Assignment size multiplier for IMAGE_SKELETONS_FROM_BOXES tasks"
237+
"Assignment size multiplier for image_skeletons_from_boxes tasks"
238238

239239
min_roi_size_w = int(os.environ.get("MIN_ROI_SIZE_W", 350))
240-
"Minimum absolute ROI size for IMAGE_BOXES_FROM_POINTS and IMAGE_SKELETONS_FROM_BOXES tasks"
240+
"Minimum absolute ROI size for image_boxes_from_points and image_skeletons_from_boxes tasks"
241241

242242
min_roi_size_h = int(os.environ.get("MIN_ROI_SIZE_H", 300))
243-
"Minimum absolute ROI size for IMAGE_BOXES_FROM_POINTS and IMAGE_SKELETONS_FROM_BOXES tasks"
243+
"Minimum absolute ROI size for image_boxes_from_points and image_skeletons_from_boxes tasks"
244244

245245

246246
class HumanAppConfig:

packages/examples/cvat/exchange-oracle/src/core/tasks/simple.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44

55
import datumaro as dm
66

7-
# These details are relevant for IMAGE_POINTS and IMAGE_BOXES tasks
7+
# These details are relevant for image_points and image_boxes tasks
88

99

1010
class TaskMetaLayout:

packages/examples/cvat/exchange-oracle/src/core/types.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -39,11 +39,11 @@ class JobStatuses(str, Enum, metaclass=BetterEnumMeta):
3939

4040

4141
class TaskTypes(str, Enum, metaclass=BetterEnumMeta):
42-
image_label_binary = "IMAGE_LABEL_BINARY"
43-
image_points = "IMAGE_POINTS"
44-
image_boxes = "IMAGE_BOXES"
45-
image_boxes_from_points = "IMAGE_BOXES_FROM_POINTS"
46-
image_skeletons_from_boxes = "IMAGE_SKELETONS_FROM_BOXES"
42+
image_label_binary = "image_label_binary"
43+
image_points = "image_points"
44+
image_boxes = "image_boxes"
45+
image_boxes_from_points = "image_boxes_from_points"
46+
image_skeletons_from_boxes = "image_skeletons_from_boxes"
4747

4848

4949
class CvatLabelTypes(str, Enum, metaclass=BetterEnumMeta):

packages/examples/cvat/exchange-oracle/src/handlers/job_creation.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -137,7 +137,7 @@ def add_message(self, message: str, *, sample_id: str, sample_subset: str):
137137

138138
class SimpleTaskBuilder:
139139
"""
140-
Handles task creation for IMAGE_POINTS and IMAGE_BOXES task types
140+
Handles task creation for image_points and image_boxes task types
141141
"""
142142

143143
def __init__(self, manifest: TaskManifest, escrow_address: str, chain_id: int) -> None:

packages/examples/cvat/exchange-oracle/tests/utils/setup_cvat.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77
from sqlalchemy.sql import select
88

99
from src.core.config import CvatConfig
10+
from src.core.types import ProjectStatuses, TaskTypes
1011
from src.db import SessionLocal
1112
from src.models.cvat import Assignment, Job, Project, Task, User
1213

@@ -31,8 +32,8 @@ def add_cvat_project_to_db(cvat_id: int) -> str:
3132
id=project_id,
3233
cvat_id=cvat_id,
3334
cvat_cloudstorage_id=1,
34-
status="annotation",
35-
job_type="IMAGE_LABEL_BINARY",
35+
status=ProjectStatuses.annotation.value,
36+
job_type=TaskTypes.image_label_binary.value,
3637
escrow_address="0x86e83d346041E8806e352681f3F14549C0d2BC67",
3738
chain_id=80002,
3839
bucket_url="https://test.storage.googleapis.com/",

packages/examples/cvat/recording-oracle/src/core/tasks/simple.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44

55
import datumaro as dm
66

7-
# These details are relevant for IMAGE_POINTS and IMAGE_BOXES tasks
7+
# These details are relevant for image_points and image_boxes tasks
88

99

1010
class TaskMetaLayout:

packages/examples/cvat/recording-oracle/src/core/types.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -11,11 +11,11 @@ class Networks(int, Enum):
1111

1212

1313
class TaskTypes(str, Enum, metaclass=BetterEnumMeta):
14-
image_label_binary = "IMAGE_LABEL_BINARY"
15-
image_points = "IMAGE_POINTS"
16-
image_boxes = "IMAGE_BOXES"
17-
image_boxes_from_points = "IMAGE_BOXES_FROM_POINTS"
18-
image_skeletons_from_boxes = "IMAGE_SKELETONS_FROM_BOXES"
14+
image_label_binary = "image_label_binary"
15+
image_points = "image_points"
16+
image_boxes = "image_boxes"
17+
image_boxes_from_points = "image_boxes_from_points"
18+
image_skeletons_from_boxes = "image_skeletons_from_boxes"
1919

2020

2121
class OracleWebhookTypes(str, Enum):

0 commit comments

Comments
 (0)