Skip to content

Commit ae71687

Browse files
rahul003jarednielsen
authored andcommitted
Fix bug in values method (aws#67)
1 parent 6cc3604 commit ae71687

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

smdebug/core/tensor.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -244,7 +244,7 @@ def _step(self, step_num, mode=ModeKeys.GLOBAL, worker=None):
244244

245245
def values(self, mode=ModeKeys.GLOBAL, worker=None):
246246
res = {}
247-
for step in self.steps():
247+
for step in self.steps(mode=mode):
248248
res[step] = self.value(step_num=step, mode=mode, worker=worker)
249249
return res
250250

0 commit comments

Comments
 (0)