code: def clean_raw_keys(wgts: dict):
      keys = list(wgts.keys())
      for k in keys:
          t = k.split(".module")
          if f"{_rm_module(k)}_raw" in keys:
              del wgts[k]
      return wgts
  
imports: from __future__ import annotations
  from ..basics import *
  from .core import *
  from .data import *
  from .models.core import *
  from .models.awdlstm import *
  from ..callback.rnn import *
  from ..callback.progress import *
  from .models.core import _model_meta
label_desc: Use image node_id 5 for calling a global function with example usage: clean_raw_keys(wgts) and returns: wgts