Pārlūkot izejas kodu

Fix ResourceWarning: unclosed file (#295)

Signed-off-by: Mickaël Schoentgen <contact@tiger-222.fr>
Mickaël Schoentgen 6 gadi atpakaļ
vecāks
revīzija
b5a9df9135

+ 2 - 1
examples/5_DataManagement/build_an_image_dataset.py

@@ -56,7 +56,8 @@ def read_images(dataset_path, mode, batch_size):
     imagepaths, labels = list(), list()
     if mode == 'file':
         # Read dataset file
-        data = open(dataset_path, 'r').read().splitlines()
+        with open(dataset_path) as f:
+            data = f.read().splitlines()
         for d in data:
             imagepaths.append(d.split(' ')[0])
             labels.append(int(d.split(' ')[1]))

+ 2 - 1
notebooks/5_DataManagement/build_an_image_dataset.ipynb

@@ -84,7 +84,8 @@
     "    imagepaths, labels = list(), list()\n",
     "    if mode == 'file':\n",
     "        # Read dataset file\n",
-    "        data = open(dataset_path, 'r').read().splitlines()\n",
+    "        with open(dataset_path) as f:\n",
+    "            data = f.read().splitlines()\n",
     "        for d in data:\n",
     "            imagepaths.append(d.split(' ')[0])\n",
     "            labels.append(int(d.split(' ')[1]))\n",