From 518ca467df27a161ac8318a181075f556e512efd Mon Sep 17 00:00:00 2001
From: Mathieu Beligon <mathieu@feedly.com>
Date: Fri, 11 Sep 2020 20:52:30 +0200
Subject: [PATCH] [common] (twitch construction) adapt to new datasets
---
...truct_twith_datasets_from_manual_annotation.py | 15 ++++++++++-----
1 file changed, 10 insertions(+), 5 deletions(-)
diff --git a/common/research/common/scripts/construct_twith_datasets_from_manual_annotation.py b/common/research/common/scripts/construct_twith_datasets_from_manual_annotation.py
index 2306f80..767b0ca 100644
--- a/common/research/common/scripts/construct_twith_datasets_from_manual_annotation.py
+++ b/common/research/common/scripts/construct_twith_datasets_from_manual_annotation.py
@@ -1,10 +1,15 @@
from os import remove
from shutil import copy, make_archive, move, rmtree
-from research.common.constants import TWITCH_DSET_DIR, TWITCH_DSET_ROBOTS_VIEWS_DIR, TWITCH_ROBOTS_VIEWS_DIR
-from research.common.dataset.directory_roco_dataset import DirectoryROCODataset
-from research.common.dataset.roco_dataset_descriptor import make_markdown_dataset_report
-from research.common.scripts.construct_dataset_from_manual_annotation import construct_dataset_from_manual_annotations
+from research.common.constants import (TWITCH_DSET_DIR,
+ TWITCH_DSET_ROBOTS_VIEWS_DIR,
+ TWITCH_ROBOTS_VIEWS_DIR)
+from research.common.dataset.roco_dataset_descriptor import \
+ make_markdown_dataset_report
+from research.common.datasets.roco.directory_roco_dataset import \
+ DirectoryROCODataset
+from research.common.scripts.construct_dataset_from_manual_annotation import \
+ construct_dataset_from_manual_annotations
from research.common.scripts.correct_annotations import AnnotationFileCorrector
@@ -51,7 +56,7 @@ def _make_global_report():
make_markdown_dataset_report(all_twitch_dataset, all_twitch_dataset.dataset_path)
-def _get_mixed_dataset():
+def _get_mixed_dataset() -> DirectoryROCODataset:
return DirectoryROCODataset(TWITCH_DSET_ROBOTS_VIEWS_DIR, "Twitch")
--
GitLab