def run_tinymax_test()

in src/tinymax_healthcheck/tinymax_runner.py [0:0]

• 15 lines of code
• 3 McCabe index (conditional complexity)

def run_tinymax_test() -> bool:
  """Runs the tinymax test.

  Returns:
    True if the test passed, False otherwise.
  """
  print("Running tinymax training...")
  result = checker_common.run_command(_K_RUN_TINY_MAX_TEST)
  match = re.search(r"TinyMax TEST PASS!", result.stdout)
  if not match:
    print("Tinymax startup failed. Error: ", result.stderr)
    return False
  print("Tinymax test passed.")
  if INSTANCE_TYPE in (
      "a3-highgpu-8g",
      "a3-megagpu-8g",
  ):
    with open(WORKLOAD_TERMINATE_FILE, "w") as _:
      pass
  return True