use new `update_called` from metrics (#18714)

Commit

2 years ago

use new `update_called` from metrics (#18714) Co-authored-by: awaelchli <aedu.waelchli@gmail.com>