From 8a5bbc9e9ea8480e51600cb5a30a347a454fd438 Mon Sep 17 00:00:00 2001
From: Darren Govoni <darren@rapidquest.ai>
Date: Wed, 9 Mar 2022 19:21:44 -0500
Subject: [PATCH] Updates

---
 blazer/examples/example4.py | 2 +-
 blazer/hpc/gpu/__init__.py  | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/blazer/examples/example4.py b/blazer/examples/example4.py
index c65d7aa..c32b4cb 100644
--- a/blazer/examples/example4.py
+++ b/blazer/examples/example4.py
@@ -1,6 +1,6 @@
 import logging
 logging.basicConfig(
-    format="%(asctime)s : %(levelname)s : %(message)s", level=logging.DEBUG
+    format="%(asctime)s : %(levelname)s : %(message)s", level=logging.INFO
 )
 import blazer
 import numpy as np
diff --git a/blazer/hpc/gpu/__init__.py b/blazer/hpc/gpu/__init__.py
index 0d6e05b..aa70e21 100644
--- a/blazer/hpc/gpu/__init__.py
+++ b/blazer/hpc/gpu/__init__.py
@@ -122,7 +122,7 @@ def __enter__(self, *args, **kwargs) -> Any:
         logging.debug("[%s][%s] GPU Context enter",host,rank)
 
         if len(self.GPUS) == 0:
-            logging.info("I don't have any GPUS so just exiting")
+            logging.info("[%s][%s] I don't have any GPUS so just exiting",host,rank)
             return None
 
         while True: