Use with when opening a file for upload

neptune-ai · Dec 17, 2024 · bf2b4fd · bf2b4fd
1 parent f56a033
commit bf2b4fd
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/neptune_scale/sync/files/worker.py b/src/neptune_scale/sync/files/worker.py
@@ -127,7 +127,8 @@ def _do_upload(
         try:
             url = self._request_upload_url(path)
             src = local_path.open("rb") if local_path else io.BytesIO(data)  # type: ignore
-            upload_file(src, url, mime_type)
+            with src:
+                upload_file(src, url, mime_type)
 
             request_id = self._submit_attribute(attribute_path, path, timestamp)
             self._wait_for_completion(request_id)