Merge pull request #55 from tsugumi-sys/fix-arg-type

fix arg-type error

pipelines/moving_mnist_pipeline/data_loader.py

@@ -1,12 +1,12 @@
 from typing import Tuple
 
 import torch
-from torch.utils.data import DataLoader, Dataset, random_split
+from torch.utils.data import DataLoader, Dataset, Subset, random_split
 from torchvision.datasets import MovingMNIST
 
 
 class VideoPredictionDataset(Dataset):
-    def __init__(self, data: torch.Tensor, input_frames: int = 10):
+    def __init__(self, data: Subset, input_frames: int = 10):
         self.data = data
         self.input_frames = input_frames
 

pyproject.toml

@@ -30,10 +30,4 @@ split-on-trailing-comma = true
 namespace_packages = true
 ignore_missing_imports = true
 python_version = "3.11"
-disable_error_code = [
-  "arg-type",
-  "assignment",
-  "override",
-  "index",
-  "var-annotated",
-]
+disable_error_code = ["assignment", "override", "index", "var-annotated"]