Skip to content

Commit a5c6012

Browse files
committed
make black happy
1 parent 856a949 commit a5c6012

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

tests/data/huggingface/run_tf.py

+1-3
Original file line numberDiff line numberDiff line change
@@ -65,8 +65,7 @@
6565
x: train_dataset[x].to_tensor(default_value=0, shape=[None, tokenizer.model_max_length])
6666
for x in ["input_ids", "attention_mask"]
6767
}
68-
69-
68+
7069
tf_train_dataset = tf.data.Dataset.from_tensor_slices(
7170
(train_features, train_dataset["label"])
7271
).batch(args.per_device_train_batch_size)
@@ -85,7 +84,6 @@
8584
for x in ["input_ids", "attention_mask"]
8685
}
8786

88-
8987
tf_test_dataset = tf.data.Dataset.from_tensor_slices(
9088
(test_features, test_dataset["label"])
9189
).batch(args.per_device_eval_batch_size)

0 commit comments

Comments
 (0)