diff --git a/src/unitxt/standard.py b/src/unitxt/standard.py index ca396b29c..e57bd4271 100644 --- a/src/unitxt/standard.py +++ b/src/unitxt/standard.py @@ -460,19 +460,7 @@ def reset_pipeline(self): ) else: raise ValueError("num_demos must be int or List[int]") - # here for backward compatibility. to check - if isinstance(self.template, list): - self.verbalization.steps.append( - ApplyRandomTemplate( - templates=self.template, demos_field=self.demos_field - ) - ) - else: - self.verbalization.steps.append( - ApplySingleTemplate( - template=self.template, demos_field=self.demos_field - ) - ) + else: self.verbalization.steps.append( Set(