From 3766ec60d874f40441802adde87072cc80c4375f Mon Sep 17 00:00:00 2001
From: emetheni <lenakmeth@gmail.com>
Date: Fri, 12 May 2023 19:32:30 +0200
Subject: [PATCH] fix indent
---
adapter_classifier.py | 16 ++++++++--------
1 file changed, 8 insertions(+), 8 deletions(-)
diff --git a/adapter_classifier.py b/adapter_classifier.py
index 8352974..478ae6e 100644
--- a/adapter_classifier.py
+++ b/adapter_classifier.py
@@ -73,16 +73,16 @@ train_dataset = train_dataset.map(encode_batch, batched=True)
train_dataset.set_format(type="torch", columns=["input_ids", "attention_mask", "labels"])
encoded_dev_dataset = {}
- for corpus in dev_dict_dataset:
- temp = dev_dict_dataset[corpus].map(encode_batch, batched=True)
- temp.set_format(type="torch", columns=["input_ids", "attention_mask", "labels"])
- encoded_dev_dataset[corpus] = temp
+for corpus in dev_dict_dataset:
+ temp = dev_dict_dataset[corpus].map(encode_batch, batched=True)
+ temp.set_format(type="torch", columns=["input_ids", "attention_mask", "labels"])
+ encoded_dev_dataset[corpus] = temp
encoded_test_dataset = {}
- for corpus in test_dict_dataset:
- temp = test_dict_dataset[corpus].map(encode_batch, batched=True)
- temp.set_format(type="torch", columns=["input_ids", "attention_mask", "labels"])
- encoded_test_dataset[corpus] = temp
+for corpus in test_dict_dataset:
+ temp = test_dict_dataset[corpus].map(encode_batch, batched=True)
+ temp.set_format(type="torch", columns=["input_ids", "attention_mask", "labels"])
+ encoded_test_dataset[corpus] = temp
# ===============================
# ## Training params
--
GitLab