update metric.py

Enhance codefactor.
thu-coai · Jan 30, 2019 · e97fee6 · e97fee6
1 parent 46e25f2
commit e97fee6
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/contk/metric/metric.py b/contk/metric/metric.py
@@ -355,9 +355,12 @@ def forward(self, data):
 		if len(context) != len(reference) or len(context) != len(gen):
 			raise ValueError("Batch num is not matched.")
 		for i, context_sen in enumerate(context):
-			self.context_list.append(self.dataloader.multi_turn_index_to_sen(np.array(context_sen)[ :, 1:]))
-			self.reference_list.append(self.dataloader.multi_turn_index_to_sen(np.array(reference[i])[ :, 1:]))
-			self.gen_list.append(self.dataloader.multi_turn_index_to_sen(np.array(gen[i])[ :]))
+			self.context_list.append(self.dataloader.multi_turn_index_to_sen( \
+				np.array(context_sen)[ :, 1:]))
+			self.reference_list.append(self.dataloader.multi_turn_index_to_sen( \
+				np.array(reference[i])[ :, 1:]))
+			self.gen_list.append(self.dataloader.multi_turn_index_to_sen( \
+				np.array(gen[i])[ :]))
 
 	def close(self):
 		'''Return a dict which contains: