Unverified Commit 651688b2 authored by Yuge Zhang's avatar Yuge Zhang Committed by GitHub
Browse files

Replace `logging.info` with `logger.info` (#4913)

parent 0064e40e
...@@ -25,21 +25,21 @@ try: ...@@ -25,21 +25,21 @@ try:
import torch import torch
TORCH_VERSION = _minor_version_tuple(torch.__version__) TORCH_VERSION = _minor_version_tuple(torch.__version__)
except ImportError: except ImportError:
logging.info("PyTorch is not installed.") logging.getLogger(__name__).info("PyTorch is not installed.")
TORCH_VERSION = None TORCH_VERSION = None
try: try:
import pytorch_lightning import pytorch_lightning
PYTORCH_LIGHTNING_VERSION = _minor_version_tuple(pytorch_lightning.__version__) PYTORCH_LIGHTNING_VERSION = _minor_version_tuple(pytorch_lightning.__version__)
except ImportError: except ImportError:
logging.info("PyTorch Lightning is not installed.") logging.getLogger(__name__).info("PyTorch Lightning is not installed.")
PYTORCH_LIGHTNING_VERSION = None PYTORCH_LIGHTNING_VERSION = None
try: try:
import tensorflow import tensorflow
TENSORFLOW_VERSION = _minor_version_tuple(tensorflow.__version__) TENSORFLOW_VERSION = _minor_version_tuple(tensorflow.__version__)
except ImportError: except ImportError:
logging.info("Tensorflow is not installed.") logging.getLogger(__name__).info("Tensorflow is not installed.")
TENSORFLOW_VERSION = None TENSORFLOW_VERSION = None
# Serialization version check are needed because they are prone to be inconsistent between versions # Serialization version check are needed because they are prone to be inconsistent between versions
......
...@@ -425,7 +425,7 @@ class Experiment: ...@@ -425,7 +425,7 @@ class Experiment:
experiment_profile = self.get_experiment_profile() experiment_profile = self.get_experiment_profile()
experiment_profile['params'][key] = value experiment_profile['params'][key] = value
rest.put(self.port, api, experiment_profile, self.url_prefix) rest.put(self.port, api, experiment_profile, self.url_prefix)
logging.info('Successfully update %s.', key) _logger.info('Successfully update %s.', key)
def update_trial_concurrency(self, value: int): def update_trial_concurrency(self, value: int):
""" """
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment