Re-add check for dataset subcommand.

Signed-off-by: Frank Di Natale <3429989+FrankD412@users.noreply.github.com>

Author: FrankD412

tests/integration/defs/perf/utils.py

@@ -451,7 +451,7 @@ def run_ex(self,
         tmpDir = temp_wd(self.get_working_dir())
 
         cmd_str = commands.get_cmd_str(cmd_idx)
-        is_prepare_dataset_cmd = 'prepare_dataset' in cmd_str
+        is_prepare_dataset_cmd = 'prepare_dataset' in cmd_str or "dataset --output" in cmd_str
 
         # Start the timer.
         self._start_timestamp = datetime.utcnow()