Add missing import for publish_gauge_metric

Description

!537 (merged) used publish_gauge_metric in cleanup.py, but didn't import it. This MR fixes that.