avoid non-exist protobuf field

alibaba · Dec 26, 2024 · 9783bb5 · 9783bb5
1 parent 4d17473
commit 9783bb5
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/tzrec/features/feature.py b/tzrec/features/feature.py
@@ -401,7 +401,12 @@ def dense_embedding_config(
     ) -> Optional[DenseEmbeddingConfig]:
         """Get DenseEmbeddingConfig of the feature."""
         if not self.is_sparse:
-            if self.config.HasField("autodis") and self.config.HasField("mlp"):
+            if (
+                hasattr(self.config, "autodis")
+                and self.config.HasField("autodis")
+                and hasattr(self.config, "mlp")
+                and self.config.HasField("mlp")
+            ):
                 raise ValueError(
                     f"feature [{self.name}] can not be configured in\
                      both autodis and mlp embedding."