fix bug creapy config

pvq_manipulation/helper/creapy_wrapper.py

@@ -14,7 +14,9 @@ from creapy.feature_extraction.feature_extraction import _cpp, _h1_h2, _jitter,
 from sklearn.ensemble import RandomForestClassifier
 from sklearn.neural_network import MLPClassifier
 
-_CONFIG_DIR = "./creapy_config.yaml"
+_CONFIG_DIR = "./pvq_manipulation/helper/creapy_config.yaml"
+_USER_CONFIG_DIR = "./pvq_manipulation/helper/user_config.yaml"
+USER_CONFIG_DIR = "./pvq_manipulation/helper/user_config.yaml"
 
 
 def get_config() -> dict:
@@ -27,6 +29,23 @@ def get_config() -> dict:
 
     with open(_CONFIG_DIR) as config_file:
         config: dict = ruamel.yaml.safe_load(config_file.read())
+
+    with open(_USER_CONFIG_DIR) as user_config_file:
+        _user_config: dict = ruamel.yaml.safe_load(user_config_file.read())
+
+    config['USER'] = _user_config
+
+    if not isfile(USER_CONFIG_DIR):
+        return config
+
+    with open(USER_CONFIG_DIR) as user_config_file:
+        user_config: dict = ruamel.yaml.safe_load(user_config_file.read())
+
+    for key in user_config.keys():
+        if key not in config['USER']:
+            raise ValueError(f'Invalid key found in user config: {key}')
+
+    config['USER'].update(user_config)
     return config
 
 
@@ -145,7 +164,7 @@ def blockwise_feature_calculation(data: np.ndarray, sr, feature):
 
 
 def process_file(data, sample_rate: int = 16_000):
-    _config = creapy.config.get_config()
+    _config = get_config()
     user_cfg = _config['USER']
     model_cfg = _config['MODEL']
 

pvq_manipulation/helper/user_config.yaml

@@ -0,0 +1,16 @@
+audio_directory: null
+textgrid_directory: null
+csv_directory: null
+audio_start: 0
+audio_end: -1
+audio_suffix: .wav
+filename_extension: 
+textgrid_suffix: .TextGrid
+gender_model: all
+tier_name: creapy
+verbose: true
+block_size: 0.040 # seconds
+hop_size: 0.010 # in seconds
+creak_threshold: 0.75
+zcr_threshold: 0.10
+ste_threshold: 0.00001