[Ref] list comprehensions in label filtering

src/dataloading.py

@@ -131,8 +131,7 @@ def filter_labels(dataset_name, doc):
             elif d[target_key] == "no":
                 labels.append(1)
     elif test_target.is_digit():
-        for d in doc:
-            labels.append(int(d[target_key]))
+        labels = [int(d[target_key]) for d in doc]
 
     return labels