Update 2 files

- /wandb.py
- /trainer.cli.py

trainer.cli.py

@@ -6,6 +6,8 @@ from argparse import ArgumentParser
 
 from util import ConfigParser
 
+from wandb import Wandb
+
 
 
 parser = ArgumentParser(
@@ -23,6 +25,6 @@ if __name__ == '__main__':
 
     config = ConfigParser(args.config_path).config
 
-    print(f"Config: {config}")
-    print(f"Wandb: {config.wandb}")
-    print(f"Wandb.name: {config.wandb.name}")
+
+    wandb = Wandb(config)
+    print(wandb.name)

wandb.py

@@ -7,11 +7,13 @@ from util import Config, RandomCode
 
 class Wandb:
     def __init__(self, config: Config):
-        if config.entity is None or config.project is None or config.api_key is None:
+        self.__dict__ = config.__dict__
+
+        if entity is None or project is None or api_key is None:
             raise ValueError('Missing required values for either Entity, Project or API Key')
 
-        if config.name == '':  
-            config.name = 'run-' + RandomCode()
+        if name == '':  
+            name = 'run-' + RandomCode()
 
         os.environ['WANDB_API_KEY'] = api_key
         wandb.init(project, entity, name)