Method __init__

eval/run_eval_agent.py:37–41

(self, data_dir, split="test", ratio=1.0)

    }

    def __init__(self, data_dir, split="test", ratio=1.0) -> None:
        self.ratio = ratio
        self.data_dir = os.path.join(data_dir, split)
        self.episode_data = self._load_data_()
        self.data = self._split_to_steps_(self.episode_data)

    def _load_data_(self):
        valid_paths = defaultdict(list)