From 3a8c48c722f687a47180221f807bd0a59c9f616e Mon Sep 17 00:00:00 2001
From: Jordan Gong <jordan.gong@protonmail.com>
Date: Sun, 27 Dec 2020 14:43:27 +0800
Subject: Add dataset selector to config type hint, change ClipLabels typo to
 ClipViews

---
 utils/configuration.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

(limited to 'utils/configuration.py')

diff --git a/utils/configuration.py b/utils/configuration.py
index 84bd064..445badd 100644
--- a/utils/configuration.py
+++ b/utils/configuration.py
@@ -1,7 +1,9 @@
-from typing import TypedDict
+from typing import TypedDict, Optional, Union
 
 import torch
 
+from utils.dataset import ClipLabels, ClipConditions, ClipViews
+
 
 class SystemConfiguration(TypedDict):
     device: torch.device
@@ -15,6 +17,7 @@ class DatasetConfiguration(TypedDict):
     train_size: int
     num_sampled_frames: int
     discard_threshold: int
+    selector: Optional[dict[str, Union[ClipLabels, ClipConditions, ClipViews]]]
     num_input_channels: int
     frame_size: tuple[int, int]
     cache_on: bool
-- 
cgit v1.2.3