little importer bugfix

Summary: fix an untested PR with another untested PR. facebookresearch#230 (comment) Reviewed By: lw Differential Revision: D30404532 fbshipit-source-id: befa2cc64a8095780eeccb5f261d8b13802bd7b3
kevin-14 · Aug 19, 2021 · 26e23ee · 26e23ee
1 parent b4bb281
commit 26e23ee
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchbiggraph/converters/importers.py b/torchbiggraph/converters/importers.py
@@ -66,7 +66,7 @@ def read(self, path: Path):
                     rel_word = words[self.rel_col] if self.rel_col is not None else None
                     weight_word = (
                         float(words[self.weight_col])
-                        if self.weight_word is not None
+                        if self.weight_col is not None
                         else None
                     )
                     yield lhs_word, rhs_word, rel_word, weight_word