From cca888785ce5e95ae8114a6f4d4c8a7033958006 Mon Sep 17 00:00:00 2001
From: Will Constable <whc@meta.com>
Date: Tue, 21 May 2024 17:34:52 -0700
Subject: [PATCH] Update (base update)

[ghstack-poisoned]
---
 test_runner.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/test_runner.py b/test_runner.py
index dfd4a987..3fd640ab 100755
--- a/test_runner.py
+++ b/test_runner.py
@@ -5,13 +5,15 @@
 # LICENSE file in the root directory of this source tree.
 
 import argparse
+import logging
 import os
 import subprocess
 from collections import defaultdict
 from dataclasses import dataclass
 from typing import Sequence
 
-from torchtitan.logging_utils import logger
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
 
 try:
     import tomllib