def setup()

in train/datamodules/slip_detect.py [0:0]

• 25 lines of code
• 2 McCabe index (conditional complexity)

    def setup(self, stage=None):
        slip_dataset_fn = SlipVideoClip if self.video_clip else SlipVideo

        self.train_dataset = slip_dataset_fn(
            self.cfg.data.path,
            self.cfg.data.prefix,
            spatial_transform=self.cfg.data.spatial_transform,
            temporal_transform=self.cfg.data.spatial_transform,
            target_transform=self.cfg.data.target_transform,
            subset="train",
            frame_duration=self.cfg.data.frame_duration,
        )

        self.val_dataset = slip_dataset_fn(
            self.cfg.data.path,
            self.cfg.data.prefix,
            spatial_transform=self.cfg.data.spatial_transform,
            temporal_transform=self.cfg.data.spatial_transform,
            target_transform=self.cfg.data.target_transform,
            subset="validation",
            frame_duration=self.cfg.data.frame_duration,
        )

        self.dataset_len = len(self.train_dataset) + len(self.val_dataset)

        _log.info(
            f"Total dataset size: {self.dataset_len}, "
            + f"train {len(self.train_dataset)}, val {len(self.val_dataset)}"
        )