download empty model in setup

for unit testing
BirdVox · Dec 3, 2018 · 304c3b8 · 304c3b8
1 parent 2a6bcfe
commit 304c3b8
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -12,7 +12,8 @@
 model_dir = os.path.join('birdvoxdetect', 'models')
 suffixes = [
     'pcen_cnn',
-    'pcen_cnn_adaptive-threshold-T1800'
+    'pcen_cnn_adaptive-threshold-T1800',
+    'empty' # for unit tests
 ]
 weight_files = ['birdvoxdetect_{}.h5'.format(suffix) for suffix in suffixes]
 base_url = 'https://github.com/BirdVox/birdvoxdetect/raw/models/'