wandb logger test fix (#572)

bfineran · web-flow · commit 9f05de5819bb · 2022-02-25T09:45:23.000-05:00
diff --git a/src/sparseml/pytorch/utils/logger.py b/src/sparseml/pytorch/utils/logger.py
@@ -402,6 +402,8 @@ def __init__(
 
         if init_kwargs:
             wandb.init(**init_kwargs)
+        else:
+            wandb.init()
 
     def _log_lambda(
         self,
diff --git a/tests/sparseml/pytorch/utils/test_logger.py b/tests/sparseml/pytorch/utils/test_logger.py
@@ -43,7 +43,7 @@
             )
             or True
         ),
-        *([WANDBLogger] if WANDBLogger.available() else []),
+        *([WANDBLogger()] if WANDBLogger.available() else []),
         SparsificationGroupLogger(
             lambda_func=lambda tag, value, values, step, wall_time: logging.info(
                 f"{tag}, {value}, {values}, {step}, {wall_time}"

Original file line number	Diff line number	Diff line change
`@@ -43,7 +43,7 @@`
`43`	`43`	`)`
`44`	`44`	`or True`
`45`	`45`	`),`
`46`		`- *([WANDBLogger] if WANDBLogger.available() else []),`
	`46`	`+ *([WANDBLogger()] if WANDBLogger.available() else []),`
`47`	`47`	`SparsificationGroupLogger(`
`48`	`48`	`lambda_func=lambda tag, value, values, step, wall_time: logging.info(`
`49`	`49`	`f"{tag}, {value}, {values}, {step}, {wall_time}"`