Update file wandb.py

wandb.py

@@ -8,31 +8,20 @@ from util import Config, RandomCode
 class Wandb:
     def __init__(self, config: Config):
         self.__dict__ = dict(config.__dict__)
-        locals().update(self.__dict__)
 
-        ewoirjwer='wrqwer'
-
-        print(f"locals: {locals()}")
-        print(f"self: {self}")
-        print(f'self dict: {dict(self.__dict__)}')
-        
-        #print(f"config: {config}")
-        #print(f'config dict: {dict(config.__dict__)}')
-
-
-        if entity == '' or project == '' or api_key == '':
+        if self.entity == '' or self.project == '' or self.api_key == '':
             raise ValueError('Missing required values for either Entity, Project or API Key')
 
-        if name == '':  
-            name = 'run-' + RandomCode()
+        if self.name == '':  
+            self.name = 'run-' + RandomCode()
 
-        os.environ['WANDB_API_KEY'] = api_key
-        wandb.init(project, entity, name)
+        os.environ['WANDB_API_KEY'] = self.api_key
+        wandb.init(self.project, self.entity, self.name)
 
 
-    #def __call__(self, args):
-    #    wandb.log(args)
+    def __call__(self, args):
+        wandb.log(args)
 
 
-    #def __del__(self):
-    #    wandb.finish()
+    def __del__(self):
+        wandb.finish()