Merge pull request #155 from dmwm/nikodemas-patch-1

Update wma_agent_count_to_opensearch.py
dmwm · Jan 11, 2024 · 754c3d5 · 754c3d5
2 parents a23bb8d + 1fe1e53
commit 754c3d5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/python/CMSSpark/wma_agent_count_to_opensearch.py b/src/python/CMSSpark/wma_agent_count_to_opensearch.py
@@ -61,7 +61,7 @@ def udf_step_extract(row):
     site_name, count = "UNKNOWN", 0
 
     result = {'host': host, 'ts': ts, 'wmaid': wmaid, 'wmats': wmats}
-    if 'steps' in row:
+    if 'steps' in row and row['steps']:
         for step in row['steps']:
             count += 1
             if step['site'] is not None: