Update 2 files

- /util.py
- /wandb.py

util.py

@@ -1,6 +1,18 @@
 import json
+import math
+import random
 
 
+def RandomCode():
+    code = '';
+    chars = '0123456789abcdef'
+    count = 8;
+
+    for i in range(0, count):
+        code += chars[math.floor(random.randrange(len(chars)))]
+
+    return code
+
 
 class Config:
     def __init__(self, data):

wandb.py

@@ -0,0 +1,25 @@
+import wandb
+
+
+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:
+            raise ValueError('Missing required values for either Entity, Project or API Key')
+
+        if config.name == '':  
+            config.name = 'run-' + RandomCode()
+
+        os.environ['WANDB_API_KEY'] = api_key
+        wandb.init(project, entity, name)
+
+
+    def __call__(self, args):
+        wandb.log(args)
+
+
+    def __del__(self):
+        wandb.finish()