From 0e69a9808d0c9a5827458f249455d970ae94cf2a Mon Sep 17 00:00:00 2001
From: Marc Bonnici <marc.bonnici@arm.com>
Date: Thu, 19 Jul 2018 11:11:43 +0100
Subject: [PATCH] commands/record: Fix argument validation

When ensuring that at least one stage for a workload recording was
present there was a missing check to see if recording for a workload was
specified.
---
 wa/commands/revent.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/wa/commands/revent.py b/wa/commands/revent.py
index 8314915a..e906cb6e 100644
--- a/wa/commands/revent.py
+++ b/wa/commands/revent.py
@@ -100,7 +100,7 @@ class RecordCommand(Command):
                                   args.teardown or args.all):
             self.logger.error("Cannot specify a recording stage without a Workload")
             sys.exit()
-        if not (args.all or args.teardown or args.extract_results or args.run or args.setup):
+        if args.workload and not any([args.all, args.teardown, args.extract_results, args.run, args.setup]):
             self.logger.error("Please specify which workload stages you wish to record")
             sys.exit()