code: def __init__(
      self,
      env: Gym,
      agent: Agent,
      num_episodes: int,
      seed: int = 0,
      max_steps: Optional[int] = None,
  ):
      self.env = env
      self.agent = agent
      self.num_episodes = num_episodes
      self.seed = seed
      self.max_steps = max_steps
  
file_path: ReAgent/reagent/gym/datasets/episodic_dataset.py
imports: import logging
  from typing import Optional
  import torch
  from reagent.gym.agents.agent import Agent
  from reagent.gym.envs.gym import Gym
  from reagent.gym.runners.gymrunner import run_episode
label_desc: Use image node_id 1 to create a new EpisodicDataset object from inherited base classes: torch.utils.data with example: obj = EpisodicDataset(env, agent, num_episodes, seed, max_steps)