File tree Expand file tree Collapse file tree 2 files changed +17
-2
lines changed Expand file tree Collapse file tree 2 files changed +17
-2
lines changed Original file line number Diff line number Diff line change 1
1
from enum import Enum
2
2
3
+ from labelbox .utils import camel_case
4
+
3
5
4
6
class MediaType (Enum ):
5
7
Audio = "AUDIO"
@@ -23,7 +25,7 @@ class MediaType(Enum):
23
25
LLM = "LLM"
24
26
25
27
@classmethod
26
- def _missing_ (cls , value : str ):
28
+ def _missing_ (cls , value ):
27
29
"""Handle missing null data types for projects
28
30
created without setting allowedMediaType
29
31
Handle upper case names for compatibility with
@@ -32,8 +34,18 @@ def _missing_(cls, value: str):
32
34
if value is None :
33
35
return cls .Unknown
34
36
37
+ def matches (value , name ):
38
+ value_upper = value .upper ()
39
+ name_upper = name .upper ()
40
+ value_underscore = value .replace ("-" , "_" )
41
+ camel_case_value = camel_case (value_underscore )
42
+
43
+ return (value_upper == name_upper or
44
+ value_underscore .upper () == name_upper or
45
+ camel_case_value .upper () == name_upper )
46
+
35
47
for name , member in cls .__members__ .items ():
36
- if name . upper () == value . upper ( ):
48
+ if matches ( value , name ):
37
49
return member
38
50
39
51
@classmethod
Original file line number Diff line number Diff line change @@ -82,3 +82,6 @@ def test_request_labeling_service_dashboard_filters(requested_labeling_service):
82
82
]
83
83
assert len (labeling_service_dashboard ) == 0
84
84
assert labeling_service_dashboard == []
85
+ labeling_service_dashboard = project .client .get_labeling_service_dashboards (
86
+ ).get_one ()
87
+ assert labeling_service_dashboard
You can’t perform that action at this time.
0 commit comments