Skip to content

Commit 41fdc80

Browse files
owen-tddavydenko
authored andcommitted
Pass iteration_number to metrics.log_metric as keyword argument. Fix bug where it was being passed to the timestamp positional argument. (#62)
Overriding CI fail due to urgency and clear fix.
1 parent d28dc42 commit 41fdc80

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

smdebug/core/hook.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -569,7 +569,8 @@ def _write_scalars(self):
569569
write_tb = scalar_obj.write_tb
570570
write_event = scalar_obj.write_event
571571
if self.metrics_writer and sm_metric:
572-
self.metrics_writer.log_metric(scalar_name, scalar_val, self.mode_steps[self.mode])
572+
self.metrics_writer.log_metric(scalar_name, scalar_val,
573+
iteration_number=self.mode_steps[self.mode])
573574
if write_tb:
574575
tb_writer = self._maybe_get_tb_writer()
575576
if tb_writer:

0 commit comments

Comments
 (0)