Skip to content

Commit

Permalink
Make test_runner use separate logger with default INFO
Browse files Browse the repository at this point in the history
previous change to use logging from torchtitan caused stdout not
to show up.

ghstack-source-id: 30a77c59ba68043ffa844be0443d5351d9584fab
Pull Request resolved: #352
  • Loading branch information
wconstab committed May 22, 2024
1 parent 9954e19 commit f47f442
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion test_runner.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit f47f442

Please sign in to comment.