fix(param): exclude_cls param

2023-09-13 15:40:04 -07:00 · 2023-09-13 15:40:04 -07:00 · 35038708e4
--- a/archai/discrete_search/evaluators/nlp/parameters.py
+++ b/archai/discrete_search/evaluators/nlp/parameters.py
@ -29,7 +29,7 @@ class NonEmbeddingParamsProxy(ModelEvaluator):

        """

-        self.exclude_cls = [nn.Embedding] or exclude_cls
+        self.exclude_cls = exclude_cls or [nn.Embedding] 
        self.trainable_only = trainable_only

    @overrides