fix(param): exclude_cls param

microsoft · Sep 13, 2023 · 3503870 · 3503870
1 parent 95d6e19
commit 3503870
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/archai/discrete_search/evaluators/nlp/parameters.py b/archai/discrete_search/evaluators/nlp/parameters.py
@@ -29,7 +29,7 @@ def __init__(self, exclude_cls: Optional[List[nn.Module]] = None, trainable_only
 
         """
 
-        self.exclude_cls = [nn.Embedding] or exclude_cls
+        self.exclude_cls = exclude_cls or [nn.Embedding] 
         self.trainable_only = trainable_only
 
     @overrides