Update file logger.py

logger.py

@@ -1,4 +1,4 @@
-import wandb as wb
+import wandb
 import os
 
 from util import Config, RandomCode
@@ -9,9 +9,6 @@ class Wandb:
     def __init__(self, config: Config):
         self.__dict__ = dict(config.__dict__)
 
-        #print(f"self.wandb: {self.wandb}")
-        #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')
 
@@ -19,15 +16,13 @@ class Wandb:
             self.name = 'run-' + RandomCode()
 
         os.environ['WANDB_API_KEY'] = self.api_key
-        self.init()
+        wandb.init(self.project, self.entity, self.name)
 
 
-    def init(self):
-        wb.init(self.project, self.entity, self.name)
 
     def __call__(self, args):
-        wb.log(args)
+        wandb.log(args)
 
 
     def __del__(self):
-        wb.finish()
+        wandb.finish()