class_docstr: Info to generate a TPU.
class_docstr_tok: ["Info","to","generate","a","TPU","."]
docstr: Parses a CLI string of the form COUNTxGPUType into a proper GPU spec
  instance.
docstr_tok: ["Parses","a","CLI","string","of","the","form","COUNTxGPUType","into","a","proper","GPU","spec","instance","."]
code: def parse_arg(s: str, **kwargs) -> "TPUSpec":
      """Parses a CLI string of the form COUNTxGPUType into a proper GPU spec
      instance.
  
      """
      tpu, count = parse_accelerator_arg(
          s, "TPU", TPUSpec._error_suffix, **kwargs
      )
      return TPUSpec(tpu, count)
  
imports: import argparse
  from enum import Enum
  from typing import Callable, List, NamedTuple, Optional, Set, Union
  import caliban.util
label_desc: Use image node_id 1 for calling the TPUSpec obj's underlying member method code with example usage: obj.parse_arg(s) and returns: TPUSpec