fix logic

kermitt2 · Oct 26, 2024 · 6fbd52f · 6fbd52f
1 parent 394aaa5
commit 6fbd52f
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/delft/sequenceLabelling/wrapper.py b/delft/sequenceLabelling/wrapper.py
@@ -108,10 +108,11 @@ def __init__(self,
         word_emb_size = 0
         self.embeddings = None
         self.model_local_path = None
-        if wandb_config is not None and 'project' not in wandb_config:
-            self.wandb_config = wandb_config
-        else:
-            raise ValueError("The wandb_config should be a dictionary with at least the string parameter 'project'. ")
+        if wandb_config is not None:
+            if 'project' not in wandb_config:
+                self.wandb_config = wandb_config
+            else:
+                raise ValueError("The wandb_config should be a dictionary with at least the string parameter 'project'. ")
 
         self.registry = load_resource_registry("delft/resources-registry.json")