Add dataset selector to config type hint, change ClipLabels typo to ClipViews

1 files changed, 4 insertions, 1 deletions

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