Updated code to match new rules

[integration/test.git] / tools / clustering / cluster-monitor / timed_isolation.py

diff --git a/tools/clustering/cluster-monitor/timed_isolation.py b/tools/clustering/cluster-monitor/timed_isolation.py
index 8492575afcf572db7ffda166a8f5cc455be378d6..8406e592effe3499167a7ac65ec4420ed7a96c60 100644 (file)
--- a/tools/clustering/cluster-monitor/timed_isolation.py
+++ b/tools/clustering/cluster-monitor/timed_isolation.py
@@ -32,11 +32,19 @@ Usage:python timed_isolation.py [controller to be isolated]  [duration of isolat
 """
 
 import sys
-sys.path.append('../../../csit/libraries')
-import UtilLibrary
-import json
 import time
 
+
+def import_utility_modules():
+    global UtilLibrary, json
+    import sys
+    sys.path.append('../../../csit/libraries')
+    import UtilLibrary
+    import json
+
+
+import_utility_modules()
+
 try:
     with open('cluster.json') as cluster_file:
         data = json.load(cluster_file)