Fix crash when workloadtype isn't specified at cli

nimbusproject · May 30, 2011 · ee60e2c · ee60e2c
1 parent 81b4814
commit ee60e2c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/python/epumgmt/main/em_core_generategraph.py b/src/python/epumgmt/main/em_core_generategraph.py
@@ -847,10 +847,11 @@ def _generate_controller(workloadtype, \
 def generate_graph(p, c, m, run_name):
     graphname = p.get_arg_or_none('graphname')
     graphtype = p.get_arg_or_none('graphtype')
-    workloadtype = p.get_arg_or_none('workloadtype').lower()
+    workloadtype = p.get_arg_or_none('workloadtype')
     if not workloadtype:
         c.log.error('Expecting workloadtype to be specified.')
         return
+    workloadtype = workloadtype.lower()
 
     node_events = NodeEvents(p, c, m, run_name)
     controller_events = ControllerEvents(p, c, m, run_name)