From 46acac10a2b516d9f6d70b7e6929c565f394db77 Mon Sep 17 00:00:00 2001
From: Kalyan Chakravarthy <chakravarthik27@gmail.com>
Date: Fri, 3 May 2024 20:21:17 +0530
Subject: [PATCH] Refactor RandomAge class to improve code readability and
 maintainability

---
 langtest/transform/robustness.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/langtest/transform/robustness.py b/langtest/transform/robustness.py
index 990ece8ca..7c6671ecb 100644
--- a/langtest/transform/robustness.py
+++ b/langtest/transform/robustness.py
@@ -1680,12 +1680,12 @@ def randomize_ages(text):
             for i in range(count):
                 if isinstance(sample, str):
                     s, _ = randomize_ages(sample)
-                    perturbed_samples.append(s)
                 else:
                     s = deepcopy(sample)
                     s.test_case, transformations = randomize_ages(s.original)
                     if s.task in ("ner", "text-classification"):
                         s.transformations = transformations
                     s.category = "robustness"
+                perturbed_samples.append(s)
 
         return perturbed_samples