chore(Linter): reformatted code with black (#211)

* chore(Linter): reformatted code with black

* Create black.yaml

wandb_setup.py

@@ -4,14 +4,15 @@ import os
 
 def get_wandb():
     from dotenv import load_dotenv
+
     load_dotenv()
-    
-    ''' 0. Login to Weights and Biases '''
-    wsb_token = os.environ.get('WANDB_API_KEY')
-    if wsb_token: 
+
+    """ 0. Login to Weights and Biases """
+    wsb_token = os.environ.get("WANDB_API_KEY")
+    if wsb_token:
         wandb.login(key=wsb_token)
         return wandb
-    else: return None #wandb.login()
-    
-    # return wandb
+    else:
+        return None  # wandb.login()
 
+    # return wandb