Merge pull request #92 from BirdVox/fix-90

Fix 90
BirdVox · Jul 12, 2021 · e96eeff · e96eeff
2 parents 5ad058e + 3b2512a
commit e96eeff
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/birdvoxdetect/core.py b/birdvoxdetect/core.py
@@ -444,7 +444,7 @@ def process_file(
             consistent_pred_dict, json_dict = classify_species(
                 classifier, chunk_pcen, th_peak_loc, taxonomy
             )
-            if row is None:
+            if consistent_pred_dict is None:
                 none_peak_ids.append(peak_id)
                 continue
             rows.append(
@@ -689,7 +689,7 @@ def process_file(
             consistent_pred_dict, json_dict = classify_species(
                 classifier, chunk_pcen, th_peak_loc, taxonomy
             )
-            if row is None:
+            if consistent_pred_dict is None:
                 none_peak_ids.append(peak_id)
                 continue
             rows.append(
@@ -942,7 +942,7 @@ def process_file(
                 consistent_pred_dict, json_dict = classify_species(
                     classifier, chunk_pcen, th_peak_loc, taxonomy
                 )
-                if row is None:
+                if consistent_pred_dict is None:
                     none_peak_ids.append(peak_id)
                     continue
                 rows.append(

diff --git a/birdvoxdetect/version.py b/birdvoxdetect/version.py
@@ -1,2 +1,2 @@
 short_version = "0.6"
-version = "0.6.0a3"
+version = "0.6.0a5"