Skip to content

Commit 5bb41e2

Browse files
authored
Update utils.py
1 parent d27512d commit 5bb41e2

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

utils.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -64,7 +64,6 @@ def load_and_cache_clone_data(args, filename, pool, tokenizer, split_tag, is_sam
6464
logger.info("Create cache data into %s", cache_fn)
6565
tuple_examples = [(example, idx, tokenizer, args) for idx, example in enumerate(examples)]
6666
features = pool.map(convert_clone_examples_to_features, tqdm(tuple_examples, total=len(tuple_examples)))
67-
# features = [convert_clone_examples_to_features(x) for x in tuple_examples]
6867
all_source_ids = torch.tensor([f.source_ids for f in features], dtype=torch.long)
6968
all_labels = torch.tensor([f.label for f in features], dtype=torch.long)
7069
data = TensorDataset(all_source_ids, all_labels)
@@ -149,7 +148,7 @@ def get_filenames(data_root, task, sub_task, split=''):
149148

150149
def read_examples(filename, data_num, task):
151150
read_example_dict = {
152-
'summarize': read_summarize_examples, # read_summarize_examples, read_summarize_indent_examples
151+
'summarize': read_summarize_examples,
153152
'refine': read_refine_examples,
154153
'translate': read_translate_examples,
155154
'concode': read_concode_examples,

0 commit comments

Comments
 (0)