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