test_prototype_datasets_builtin: Properly close all streams (#7403)

Co-authored-by: Philip Meier <[email protected]>
pytorch · Mar 9, 2023 · 4a7def8 · 4a7def8
1 parent 6218dc3
commit 4a7def8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/test_prototype_datasets_builtin.py b/test/test_prototype_datasets_builtin.py
@@ -123,7 +123,7 @@ def log_session_streams(self):
     def test_stream_closing(self, log_session_streams, dataset_mock, config):
         def make_msg_and_close(head):
             unclosed_streams = []
-            for stream in StreamWrapper.session_streams.keys():
+            for stream in list(StreamWrapper.session_streams.keys()):
                 unclosed_streams.append(repr(stream.file_obj))
                 stream.close()
             unclosed_streams = "\n".join(unclosed_streams)