def _setup(self, _stage: str | None = None) -> None:
        """Set up train and test datasets.

        Args:
            _stage (str | None): Stage for Lightning. Can be "fit" or "test".
        """
        self.train_data = UCSDpedDataset(
            clip_length_in_frames=self.clip_length_in_frames,
            frames_between_clips=self.frames_between_clips,
            target_frame=self.target_frame,
            root=self.root,
            category=self.category,
            split=Split.TRAIN,
        )

        self.test_data = UCSDpedDataset(
            clip_length_in_frames=self.clip_length_in_frames,
            frames_between_clips=self.frames_between_clips,
            target_frame=self.target_frame,
            root=self.root,
            category=self.category,
            split=Split.TEST,
        )
Set up train and test datasets.

        Args:
            _stage (str | None): Stage for Lightning. Can be "fit" or "test".
        
src/anomalib/data/datamodules/video/ucsd_ped.py