From d285405977e66bd5c1cb835867d712ab1229579e Mon Sep 17 00:00:00 2001 From: Shivam Sharma Date: Fri, 29 Nov 2024 13:52:36 +0100 Subject: [PATCH] Fix for pipeline --- dicee/query_generator.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/dicee/query_generator.py b/dicee/query_generator.py index ba68ad6d..c0c02e41 100644 --- a/dicee/query_generator.py +++ b/dicee/query_generator.py @@ -6,8 +6,6 @@ import pickle from copy import deepcopy from .static_funcs import save_pickle, load_pickle -import re - class QueryGenerator: def __init__(self, train_path, val_path: str = None, test_path: str = None, ent2id: Dict = None, rel2id: Dict = None,