Skip to content

Commit

Permalink
Remove logging code moved to hivemind
Browse files Browse the repository at this point in the history
  • Loading branch information
borzunov committed May 7, 2023
1 parent 16910cf commit a4e362d
Showing 1 changed file with 0 additions and 18 deletions.
18 changes: 0 additions & 18 deletions src/petals/utils/logging.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,31 +4,13 @@
from hivemind.utils import logging as hm_logging


def in_jupyter() -> bool:
"""Check if the code is run in Jupyter or Colab"""

try:
__IPYTHON__
return True
except NameError:
return False


def initialize_logs():
"""Initialize Petals logging tweaks. This function is called when you import the `petals` module."""

# Env var PETALS_LOGGING=False prohibits Petals do anything with logs
if os.getenv("PETALS_LOGGING", "True").lower() in ("false", "0"):
return

if in_jupyter():
os.environ["HIVEMIND_COLORS"] = "True"
importlib.reload(hm_logging)

# Remove log handlers from previous import of hivemind.utils.logging and extra handlers on Colab
hm_logging.get_logger().handlers.clear()
hm_logging.get_logger("hivemind").handlers.clear()

hm_logging.use_hivemind_log_handler("in_root_logger")

# We suppress asyncio error logs by default since they are mostly not relevant for the end user,
Expand Down

0 comments on commit a4e362d

Please sign in to comment.