Update nvtabular/ops/categorify.py

NVIDIA-Merlin · Jul 14, 2021 · 4902361 · 4902361
1 parent e6536dc
commit 4902361
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/nvtabular/ops/categorify.py b/nvtabular/ops/categorify.py
@@ -228,8 +228,8 @@ def __init__(
         # Only support two kinds of multi-column encoding
         if encode_type not in ("joint", "combo"):
             raise ValueError(f"encode_type={encode_type} not supported.")
-        if encode_type == "joint" and vocabs is not None:
-            raise ValueError("Passing in vocabs is not supported with a joint encoding.")
+        if encode_type == "combo" and vocabs is not None:
+            raise ValueError("Passing in vocabs is not supported with a combo encoding.")
 
         # Other self-explanatory initialization
         super().__init__()