From 8e3b24b9caf16ef2c40990229db6fe9468ccfa6c Mon Sep 17 00:00:00 2001 From: Chimezie Iwuanyanwu Date: Sat, 2 Dec 2023 17:40:22 -0600 Subject: [PATCH] Up-directory stepcovnet.common --- training_data_collection.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/training_data_collection.py b/training_data_collection.py index 34a5d74..534195f 100644 --- a/training_data_collection.py +++ b/training_data_collection.py @@ -9,8 +9,7 @@ import joblib import psutil -from stepcovnet import utils, data, sample_collection_helper -from stepcovnet.parameters import CONFIG, VGGISH_CONFIG +from stepcovnet import utils, data, sample_collection_helper, parameters def build_all_metadata(**kwargs): @@ -194,7 +193,7 @@ def training_data_collection( ) multi = True if multi_int == 1 else False - config = VGGISH_CONFIG if type_int == 1 else CONFIG + config = parameters.VGGISH_CONFIG if type_int == 1 else parameters.CONFIG limit = max(-1, limit) # defaulting negative inputs to -1 cores = psutil.cpu_count(logical=False) if cores < 0 else cores distributed = True if distributed_int == 1 else False