Fix suggested by @bhadreshpsavani (huggingface#11660)

Iwontbecreative · Jul 15, 2021 · bb35a9e · bb35a9e
1 parent 03d41cf
commit bb35a9e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/tensorflow/text-classification/run_text_classification.py b/examples/tensorflow/text-classification/run_text_classification.py
@@ -522,7 +522,7 @@ def preprocess_function(examples):
 
     # region Prediction losses
     # This section is outside the scope() because it's very quick to compute, but behaves badly inside it
-    if "label" in datasets["test"].features:
+    if "test" in datasets and "label" in datasets["test"].features:
         print("Computing prediction loss on test labels...")
         labels = datasets["test"]["label"]
         loss = float(loss_fn(labels, predictions).numpy())