Fix tests

HumanSignal · Oct 15, 2024 · 76b8f65 · 76b8f65
1 parent 5f872ae
commit 76b8f65
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 8 deletions.
diff --git a/src/label_studio_sdk/label_interface/control_tags.py b/src/label_studio_sdk/label_interface/control_tags.py
@@ -773,6 +773,11 @@ class DateTimeTag(ControlTag):
     _value_class: Type[DateTimeValue] = DateTimeValue
     _label_attr_name: str = "datetime"
 
+    def _label_simple(self, to_name: Optional[str] = None, *args, **kwargs) -> Region:
+        # TODO: temporary fix to force datetime to be a string
+        kwargs['datetime'] = kwargs['datetime'][0]
+        return super()._label_simple(to_name, *args, **kwargs)
+
     def to_json_schema(self):
         """
         Converts the current DateTimeTag instance into a JSON Schema.

diff --git a/tests/custom/test_interface/test_control_tags.py b/tests/custom/test_interface/test_control_tags.py
@@ -36,7 +36,7 @@ def test_validate():
 def test_textarea_label():
     conf = LabelInterface(c.TEXTAREA_CONF)
 
-    region = conf.get_control(c.FROM_NAME).label(text=["Hello", "World"])
+    region = conf.get_control(c.FROM_NAME).label(label=["Hello", "World"])
 
 
 def test_label_with_choices():

diff --git a/tests/custom/test_interface/test_control_tags_label.py b/tests/custom/test_interface/test_control_tags_label.py
@@ -31,20 +31,20 @@
 
     (OT.ImageTag, CT.RectangleTag, { "x": 10, "y": 10, "width": 10, "height": 10 }, { "x": 10.0, "y": 10.0, "width": 10.0, "height": 10.0, "rotation": 0 }),
     (OT.ImageTag, CT.RectangleLabelsTag, { "x": 10, "y": 10, "width": 10, "height": 10, "label": c.LABEL1 }, { "x": 10.0, "y": 10.0, "width": 10.0, "height": 10.0, "rotation": 0, "rectanglelabels": [ c.LABEL1 ] }),
-    (OT.ImageTag, CT.TaxonomyTag, { "taxonomy": [ [ c.LABEL1 ] ] }, { "taxonomy": [ [ c.LABEL1 ] ] }),
-    (OT.ImageTag, CT.TextAreaTag, { "text": [ EX_TEXT, EX_TEXT ] }, { "text": [ EX_TEXT, EX_TEXT ] }),
+    (OT.ImageTag, CT.TaxonomyTag, { "label": [ [ c.LABEL1 ] ] }, { "taxonomy": [ [ c.LABEL1 ] ] }),
+    (OT.ImageTag, CT.TextAreaTag, { "label": [ EX_TEXT, EX_TEXT ] }, { "text": [ EX_TEXT, EX_TEXT ] }),
 
-    (OT.ImageTag, CT.RatingTag, { "rating": 3 }, { "rating": 3 }),
+    (OT.ImageTag, CT.RatingTag, { "label": 3 }, { "rating": 3 }),
 
     (OT.ImageTag, CT.BrushTag, { "rle": [2,3,3,2] }, { "rle": [2,3,3,2], "format": "rle" }),
     (OT.ImageTag, CT.BrushLabelsTag, { "rle": [2,3,3,2], "label": c.LABEL1 }, { "rle": [2,3,3,2], "format": "rle", "brushlabels": [ c.LABEL1 ] }),
 
     ## Text labeling
-    (OT.TextTag, CT.NumberTag, { "number": 5 }, { "number": 5 }),
-    (OT.TextTag, CT.DateTimeTag, { "datetime": "2024-05-07" }, { "datetime": "2024-05-07" }),
+    (OT.TextTag, CT.NumberTag, { "label": 5 }, { "number": 5 }),
+    (OT.TextTag, CT.DateTimeTag, { "label": "2024-05-07" }, { "datetime": "2024-05-07" }),
 
-    (OT.TextTag, CT.LabelsTag, { "start": 1, "end": 10, "label": c.LABEL1 }, { "start": 1, "end": 10, "labels": [ c.LABEL1 ] }),
-    (OT.TextTag, CT.LabelsTag, { "start": 1, "end": 10, "label": [ c.LABEL1, c.LABEL2 ] }, { "start": 1, "end": 10, "labels": [ c.LABEL1, c.LABEL2 ] }),
+    (OT.TextTag, CT.LabelsTag, { "label": c.LABEL1, "start": 1, "end": 10 }, { "labels": [ c.LABEL1 ], "start": 1, "end": 10 }),
+    (OT.TextTag, CT.LabelsTag, { "label": [ c.LABEL1, c.LABEL2 ], "start": 1, "end": 10 }, { "labels": [ c.LABEL1, c.LABEL2 ], "start": 1, "end": 10 }),
 
     ## Hypertext labeling
     (OT.HyperTextTag, CT.HyperTextLabelsTag, { "start": 1, "end": 10, "startOffset": 10, "endOffset": 10, "label": c.LABEL1 }, { "start": 1, "end": 10, "startOffset": 10, "endOffset": 10, "htmllabels": [ c.LABEL1 ] }),