fix warning bug

kaylode · Mar 26, 2022 · a6eba3b · a6eba3b
1 parent 57639e0
commit a6eba3b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/theseus/utilities/cuda.py b/theseus/utilities/cuda.py
@@ -23,7 +23,7 @@ def get_devices_info(device_names="0"):
 def get_device(name='cpu') -> torch.device:
     if name.startswith('cuda'):
         if not torch.cuda.is_available():
-            LOGGER.text("CUDA is not available. Using CPU...")
+            LOGGER.text("CUDA is not available. Using CPU...", level=LoggerObserver.WARN)
             name = 'cpu'
     return torch.device(name)
 

diff --git a/theseus/utilities/loggers/stdout_logger.py b/theseus/utilities/loggers/stdout_logger.py
@@ -22,7 +22,7 @@ def __init__(self, name):
 
     def log_text(self, tag, value, level=LoggerObserver.DEBUG, **kwargs):
         if level == LoggerObserver.WARN:
-            logger.warn(value)
+            logger.warning(value)
 
         if level == LoggerObserver.INFO:
             logger.info(value)