edit 'texts' to 'text' key (#499)

SEACrowd · Mar 13, 2024 · 7d7529c · 7d7529c
1 parent b57ca66
commit 7d7529c
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/seacrowd/sea_datasets/fsl_105/fsl_105.py b/seacrowd/sea_datasets/fsl_105/fsl_105.py
@@ -95,7 +95,7 @@ def _info(self) -> datasets.DatasetInfo:
                 {
                     "id": datasets.Value("string"),
                     "video_path": datasets.Value("string"),
-                    "texts": datasets.Value("string"),
+                    "text": datasets.Value("string"),
                     "labels": datasets.ClassLabel(names=self.category),
                     "metadata": {
                         "resolution": {
@@ -164,7 +164,7 @@ def _generate_examples(self, filepath: Path, split: str) -> Tuple[int, Dict]:
                 yield key, {
                     "id": str(key),
                     "video_path": os.path.join(filepath["clips"], example["vid_path"]),
-                    "texts": example["label"],
+                    "text": example["label"],
                     "labels": example["category"],
                     "metadata": {
                         "resolution": {
@@ -179,7 +179,7 @@ def _generate_examples(self, filepath: Path, split: str) -> Tuple[int, Dict]:
                 yield key, {
                     "id": str(key),
                     "video_path": os.path.join(filepath["clips"], example["vid_path"]),
-                    "texts": example["label"],
+                    "text": example["label"],
                     "metadata": {
                         "resolution": {
                             "width": vid_width,

diff --git a/seacrowd/sea_datasets/id_msvd/id_msvd.py b/seacrowd/sea_datasets/id_msvd/id_msvd.py
@@ -122,7 +122,7 @@ def _generate_examples(self, text_path: Path, video_path: Path, split: str) -> T
                 yield i, {
                     "id": str(i),
                     "video_path": str(row["video_path"]),
-                    "texts": row["text"],
+                    "text": row["text"],
                     "metadata": {
                         "resolution": {
                             "width": None,

diff --git a/seacrowd/utils/schemas/video.py b/seacrowd/utils/schemas/video.py
@@ -14,7 +14,7 @@
     {
         "id": datasets.Value("string"),
         "video_path": datasets.Value("string"),
-        "texts": datasets.Value("string"),
+        "text": datasets.Value("string"),
         "metadata": {
             "resolution": {
                 "width": datasets.Value("int64"),