Index: infra/bots/recipe_modules/flavor/ios_flavor.py |
diff --git a/infra/bots/recipe_modules/flavor/ios_flavor.py b/infra/bots/recipe_modules/flavor/ios_flavor.py |
index 806926a4b48f00f586470def19724d44afbdfbc9..7708d5dccddd1b58dc0760dbe38febbb0739c4b8 100644 |
--- a/infra/bots/recipe_modules/flavor/ios_flavor.py |
+++ b/infra/bots/recipe_modules/flavor/ios_flavor.py |
@@ -130,7 +130,8 @@ class iOSFlavorUtils(default_flavor.DefaultFlavorUtils): |
def cleanup_steps(self): |
"""Run any device-specific cleanup steps.""" |
- if self.m.vars.do_test_steps or self.m.vars.do_perf_steps: |
+ if self.m.vars.role in (self.m.builder_name_schema.BUILDER_ROLE_TEST, |
+ self.m.builder_name_schema.BUILDER_ROLE_PERF): |
self.m.run( |
self.m.step, |
name='reboot', |