Skip to content

Commit

Permalink
Remove print
Browse files Browse the repository at this point in the history
  • Loading branch information
jacobbieker committed Feb 6, 2024
1 parent 30908f1 commit fba3b03
Showing 1 changed file with 0 additions and 1 deletion.
1 change: 0 additions & 1 deletion pvnet/models/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -118,7 +118,6 @@ def flush(self) -> dict[BatchKey, list[torch.Tensor]]:
"""Concatenate all accumulated batches, return, and clear self"""
batch = {}
for k, v in self._batches.items():
print(k)
if k == BatchKey[f"{self.key_to_keep}_t0_idx"]:
batch[k] = v[0]
else:
Expand Down

0 comments on commit fba3b03

Please sign in to comment.