From 2637f538481839dfabae932a29c8970296542059 Mon Sep 17 00:00:00 2001
From: thanhvc3 <thanhvc3@vng.com.vn>
Date: Wed, 19 Jun 2024 00:20:44 +0700
Subject: [PATCH] test

---
 main.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/main.py b/main.py
index 0072b7c..9267bb9 100644
--- a/main.py
+++ b/main.py
@@ -91,6 +91,7 @@ class Main(object):
         for line in open('./data/{}/{}'.format(self.p.dataset, "relations.dict")):
             id, rel = map(str.lower, line.strip().split('\t'))
             self.rel2id[rel] = int(id)
+            rel_set.add(rel)
 
         # self.ent2id = {ent: idx for idx, ent in enumerate(ent_set)}
         # self.rel2id = {rel: idx for idx, rel in enumerate(rel_set)}