Remove stripping of step name. Replace removing only suffix step name

apache · Jul 21, 2022 · 52cbe32 · 52cbe32
1 parent f2f239a
commit 52cbe32
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/sdks/python/apache_beam/testing/load_tests/load_test_metrics_utils.py b/sdks/python/apache_beam/testing/load_tests/load_test_metrics_utils.py
@@ -92,7 +92,12 @@ def parse_step(step_name):
   Returns:
     lower case step name without namespace and step label
   """
-  return step_name.lower().replace(' ', '_').strip('step:_')
+  prefix = 'step'
+  step_name = step_name.lower().replace(' ', '_')
+  step_name = (
+      step_name[len(prefix):]
+      if prefix and step_name.startswith(prefix) else step_name)
+  return step_name.strip(':_')
 
 
 def split_metrics_by_namespace_and_name(metrics, namespace, name):