From 8c7cb8bc7453cd3e31d0f2d08fd135f832463d0d Mon Sep 17 00:00:00 2001
From: Gilchan Park <49205846+boxorange@users.noreply.github.com>
Date: Fri, 28 Jul 2023 19:40:41 -0400
Subject: [PATCH] Fixed a type in dataset.py

---
 torchdrug/data/dataset.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/torchdrug/data/dataset.py b/torchdrug/data/dataset.py
index 34e9606..e54bdca 100644
--- a/torchdrug/data/dataset.py
+++ b/torchdrug/data/dataset.py
@@ -426,7 +426,7 @@ def load_edge(self, edge_list, node_feature, node_label, node_vocab=None, inv_no
         self.node_vocab = node_vocab
         self.inv_node_vocab = inv_node_vocab
         self.label_vocab = label_vocab
-        self.inv_node_vocab = inv_label_vocab
+        self.inv_label_vocab = inv_label_vocab
 
     def _standarize_vocab(self, vocab, inverse_vocab):
         if vocab is not None: