From c9ff15ca535869b5497a359f1776a01a4670f139 Mon Sep 17 00:00:00 2001
From: Marc Bonnici <marc.bonnici@arm.com>
Date: Tue, 3 Jul 2018 13:39:53 +0100
Subject: [PATCH] wa/workloads: Fix missing imports

---
 wa/workloads/androbench/__init__.py  | 1 +
 wa/workloads/chrome/__init__.py      | 1 +
 wa/workloads/geekbench/__init__.py   | 1 +
 wa/workloads/googlemaps/__init__.py  | 2 +-
 wa/workloads/pcmark/__init__.py      | 1 +
 wa/workloads/speedometer/__init__.py | 2 +-
 wa/workloads/stress_ng/__init__.py   | 1 +
 7 files changed, 7 insertions(+), 2 deletions(-)

diff --git a/wa/workloads/androbench/__init__.py b/wa/workloads/androbench/__init__.py
index 503ea213..3417ae98 100755
--- a/wa/workloads/androbench/__init__.py
+++ b/wa/workloads/androbench/__init__.py
@@ -15,6 +15,7 @@
 import re
 
 from wa import ApkUiautoWorkload
+from wa.framework.exception import WorkloadError
 
 class Androbench(ApkUiautoWorkload):
 
diff --git a/wa/workloads/chrome/__init__.py b/wa/workloads/chrome/__init__.py
index b0257cfc..412a71d3 100644
--- a/wa/workloads/chrome/__init__.py
+++ b/wa/workloads/chrome/__init__.py
@@ -1,4 +1,5 @@
 from wa import Parameter, ApkUiautoWorkload
+from wa.framework.exception import WorkloadError
 
 
 class Chrome(ApkUiautoWorkload):
diff --git a/wa/workloads/geekbench/__init__.py b/wa/workloads/geekbench/__init__.py
index 0adad307..53a2f850 100644
--- a/wa/workloads/geekbench/__init__.py
+++ b/wa/workloads/geekbench/__init__.py
@@ -18,6 +18,7 @@ import os
 import re
 import tempfile
 import json
+from collections import defaultdict
 
 from wa import ApkUiautoWorkload, Parameter
 from wa.framework.exception import ConfigError, WorkloadError
diff --git a/wa/workloads/googlemaps/__init__.py b/wa/workloads/googlemaps/__init__.py
index 37c5ed36..32e2b984 100644
--- a/wa/workloads/googlemaps/__init__.py
+++ b/wa/workloads/googlemaps/__init__.py
@@ -1,5 +1,5 @@
 from wa import Parameter, ApkUiautoWorkload
-
+from wa.framework.exception import WorkloadError
 
 class GoogleMaps(ApkUiautoWorkload):
 
diff --git a/wa/workloads/pcmark/__init__.py b/wa/workloads/pcmark/__init__.py
index 6f1d1e95..d580e437 100755
--- a/wa/workloads/pcmark/__init__.py
+++ b/wa/workloads/pcmark/__init__.py
@@ -17,6 +17,7 @@ import re
 import zipfile
 
 from wa import ApkUiautoWorkload
+from wa.framework.exception import WorkloadError
 
 class PcMark(ApkUiautoWorkload):
 
diff --git a/wa/workloads/speedometer/__init__.py b/wa/workloads/speedometer/__init__.py
index af4e0e8b..026b9be3 100755
--- a/wa/workloads/speedometer/__init__.py
+++ b/wa/workloads/speedometer/__init__.py
@@ -16,7 +16,7 @@ import os
 import re
 
 from wa import UiautoWorkload, Parameter
-from wa.framework.exception import ValidationError
+from wa.framework.exception import ValidationError, WorkloadError
 from wa.utils.types import list_of_strs
 from wa.utils.misc import unique
 
diff --git a/wa/workloads/stress_ng/__init__.py b/wa/workloads/stress_ng/__init__.py
index b4a9d53d..21a8db29 100644
--- a/wa/workloads/stress_ng/__init__.py
+++ b/wa/workloads/stress_ng/__init__.py
@@ -18,6 +18,7 @@ import os
 import yaml
 
 from wa import Workload, Parameter, ConfigError, Executable
+from wa.framework.exception import WorkloadError
 from wa.utils.exec_control import once