Update torch_geometric/nn/nlp/txt2kg.py

Co-authored-by: Serge Panev <[email protected]>
pyg-team · Dec 19, 2024 · 401d446 · 401d446
1 parent 6d92164
commit 401d446
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torch_geometric/nn/nlp/txt2kg.py b/torch_geometric/nn/nlp/txt2kg.py
@@ -36,7 +36,7 @@ def __init__(
         if self.local_LM:
             self.initd_LM = False
         else:
-            assert NVIDIA_API_KEY != '', "Please pass NVIDIA_API_KEY or set local_small_lm flag to True"
+            assert NVIDIA_API_KEY != '', "Please pass NVIDIA_API_KEY or set local_lm flag to True"
             self.NVIDIA_API_KEY = NVIDIA_API_KEY
             self.NIM_MODEL = NVIDIA_NIM_MODEL