added different type checks

trias702 · Jun 23, 2023 · dae91fd · dae91fd
1 parent ffc8455
commit dae91fd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/nemo/collections/common/data/dataset.py b/nemo/collections/common/data/dataset.py
@@ -328,8 +328,8 @@ class CodeSwitchedDataset(IterableDataset):
 
  def __init__(
  self,
- datasets: list,
- lang_probs: Optional[list] = None,
+ datasets: List[Any],
+ lang_probs: Optional[List[float]] = None,
  shuffle: bool = True,
  min_duration: int = 4,
  max_duration: int = 20,