bugfix export confidence

BirdVox · Jan 6, 2019 · ceef4c3 · ceef4c3
1 parent 4c796ca
commit ceef4c3
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/birdvoxdetect/core.py b/birdvoxdetect/core.py
@@ -351,7 +351,8 @@ def process_file(
         # of the input.
         chunk_confidences.append(chunk_confidence)
         total_length = sum(map(len, chunk_confidences))
-        f.create_dataset('confidence', (total_length,), dtype="float32")
+        with h5py.File(confidence_path, "w") as f:
+            f.create_dataset('confidence', (total_length,), dtype="float32")
         chunk_pointer = 0
 
         # Loop over chunks.