1  def __init__(self, metagraph, model):
2    self._metagraph = metagraph
3    self.replicate_states(model.initial_state_name)
4    self.replicate_states(model.final_state_name)
5    self.update_snapshot_name("variables")
6    self.update_snapshot_name("trainable_variables")