Update file logger.py

logger.py

@@ -1,4 +1,4 @@
-import wandb
+import wandb as wb
 import os
 
 from util import Config, RandomCode
@@ -10,8 +10,7 @@ class Wandb:
         self.__dict__ = dict(config.__dict__)
 
         #print(f"self.wandb: {self.wandb}")
-
-        print(f"self.config: {self.config}")
+        #print(f"self.config: {self.config}")
 
         if self.entity == '' or self.project == '' or self.api_key == '':
             raise ValueError('Missing required values for either Entity, Project or API Key')
@@ -24,11 +23,11 @@ class Wandb:
 
 
     def init(self):
-        wandb.init(self.project, self.entity, self.name)
+        wb.init(self.project, self.entity, self.name)
 
     def __call__(self, args):
-        wandb.log(args)
+        wb.log(args)
 
 
     def __del__(self):
-        wandb.finish()
+        wb.finish()