code: def update(self, **kwargs):
      fields = vars(self).copy()
      fields.pop("_old_context", None)
      fields.update(kwargs)
  
      with _ModelContext(**fields):
          yield
  
imports: import pickle
  import threading
  import typing
  from contextlib import contextmanager
  from copy import copy
  from dataclasses import dataclass
  from pathlib import Path
  import jax
  import jax.numpy
  import numpy
  import treex
  from elegy import data, types, utils
  from elegy.callbacks import Callback, CallbackList, History, history
  from elegy.callbacks.sigint import SigIntMode
  from elegy.data import utils
  from elegy.model.model_core import ModelCore, PredStepOutput, TestStepOutput
label_desc: Use image node_id 3 for calling the _ModelContext obj's underlying member method code with example usage: obj.update() without return types