Update

[ghstack-poisoned]
pytorch · May 22, 2024 · f7c999a · f7c999a
2 parents faed51e + c76bb15
commit f7c999a
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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