fix(transformer): change getattr to hasattr

jina-ai · May 10, 2020 · e5756b3 · e5756b3
1 parent 2370dd6
commit e5756b3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/jina/executors/encoders/nlp/transformer.py b/jina/executors/encoders/nlp/transformer.py
@@ -95,7 +95,7 @@ def post_init(self):
         self.tmp_model_path = self.model_abspath if os.path.exists(self.model_abspath) else self.model_name
         self._tokenizer = self.get_tokenizer()
 
-        self._has_cls_token = True if getattr(self._tokenizer, 'cls_token', None) is not None else False
+        self._has_cls_token = hasattr(self._tokenizer, 'cls_token')
 
         self.cls_pos = 'tail' if self.model_name == 'xlnet-base-cased' else 'head'