code: def __init__(self, model: EnvironmentModel, device=None):
      self._model = model
      self._device = device
      self._model.to(self._device)
  
imports: import logging
  import gym
  import torch
  from reagent.ope.estimators.sequential_estimators import Action, ActionDistribution, ActionSpace, IPSEstimator, Model, NeuralDualDICE, RandomRLPolicy, RewardProbability, RLEstimatorInput, RLPolicy, State, StateDistribution, Transition
  from reagent.ope.utils import RunningAverage
label_desc: Use image node_id 1 to create a new ModelWrapper object from inherited base classes: Model with example: obj = ModelWrapper(model, device)