def parse_size()

in graphlearn_torch/python/utils/units.py [0:0]

• 10 lines of code
• 6 McCabe index (conditional complexity)

def parse_size(sz) -> int:
  if isinstance(sz, int):
    return sz
  if isinstance(sz, float):
    return int(sz)
  if isinstance(sz, str):
    for suf, u in sorted(UNITS.items()):
      if sz.upper().endswith(suf):
        return int(float(sz[:-len(suf)]) * u)
  raise Exception(f"invalid size: {sz}")