replace rank_zero_info with logging

Author: donglihe-hub

libmultilabel/common_utils.py

@@ -6,7 +6,6 @@
 from functools import wraps
 
 import numpy as np
-from pytorch_lightning.utilities.rank_zero import rank_zero_info
 
 
 class AttributeDict(dict):
@@ -131,14 +130,14 @@ def is_multiclass_dataset(dataset, label="label"):
 
 
 def timer(func):
-    """Emit info-level wall time only on global rank 0."""
+    """Log info-level wall time"""
 
     @wraps(func)
     def wrapper(*args, **kwargs):
         start_time = time.time()
         value = func(*args, **kwargs)
         wall_time = time.time() - start_time
-        rank_zero_info(f"{repr(func.__name__)} finished in {wall_time:.2f} seconds")
+        logging.info(f"{repr(func.__name__)} finished in {wall_time:.2f} seconds")
         return value
 
     return wrapper