code: def tokenize(lines, token_type="word"):
      """Split the lines list into word or char tokens depending on token_type."""
  
      if token_type == "word":
          return [line.split(" ") for line in lines]
      elif token_type == "char":
          return [list(line) for line in lines]
      else:
          raise ValueError("ERROR: unknown token type", token_type)
  
imports: import collections
  import random
  import re
  import jax
  import jax.numpy
  import tensorflow_datasets
  import objax
  from objax.functional import one_hot
  from objax.zoo.rnn import RNN
label_desc: Use image node_id 1 for calling a global function with example usage: tokenize(lines, token_type) and returns: unknown, unknown