from tensorflow.python.ops import variable_scope


  @contextlib.contextmanager
  def as_default(self):
    yield


def create_eager_var_store():
  if context.in_eager_mode():
    return variable_scope.EagerVariableStore()
  else:
    return DummyVariableStore()


def scheduled_sampling(hparams, problem_hparams, dp, sharded_logits, losses,
                       sharded_features, transformed_features, model):
  """Scheduled sampling."""
