code: def __exit__(self, type, value, traceback):
      print(f"{self.desc} took {time.time()-self.t:.02f}s")
  
imports: import os
  import re
  import time
  import random
  import argparse
  import torch
  from transformers import GPT2TokenizerFast
  from jaxformer.hf.codegen.modeling_codegen import CodeGenForCausalLM
label_desc: Use image node_id 3 for calling the print_time obj's underlying member method code with example usage: obj.__exit__(type, value, traceback) without return types