From 950b81d4702a4a522a98c0ad761bac54ab6430fc Mon Sep 17 00:00:00 2001 From: Mathieu Beligon <mathieu@feedly.com> Date: Sat, 28 Mar 2020 17:47:07 -0400 Subject: [PATCH] [common] (twitch dataset) update code to use full v1 twitch dataset --- .../dataset/twitch/twitch_roco_datasets.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/common/research_common/dataset/twitch/twitch_roco_datasets.py b/common/research_common/dataset/twitch/twitch_roco_datasets.py index a04bb75..368a7c4 100644 --- a/common/research_common/dataset/twitch/twitch_roco_datasets.py +++ b/common/research_common/dataset/twitch/twitch_roco_datasets.py @@ -1,15 +1,18 @@ from enum import Enum -from research_common.constants import TWITCH_DSET_ROBOTS_VIEWS_DIR +from research_common.constants import TWITCH_DSET_DIR from research_common.dataset.directory_roco_dataset import DirectoryROCODataset class TwitchROCODataset(DirectoryROCODataset, Enum): def __init__(self, competition_name: str): - super().__init__(TWITCH_DSET_ROBOTS_VIEWS_DIR / competition_name, self.name) + super().__init__(TWITCH_DSET_DIR / "v1" / competition_name, self.name) + TWITCH_470149568 = "470149568" TWITCH_470150052 = "470150052" TWITCH_470151286 = "470151286" + TWITCH_470152289 = "470152289" + TWITCH_470152730 = "470152730" TWITCH_470152838 = "470152838" TWITCH_470153081 = "470153081" - TWITCH_470152289 = "470152289" + TWITCH_470158483 = "470158483" -- GitLab