Switch order of arguments passed to prep_model_weights in train_yml.py (

#384) * Switch order of arguments passed to prep_model_weights in train_yml.py to match rest of codebase * Add explicit keywords to the funciton call
brainglobe · Feb 16, 2024 · 749af84 · 749af84
1 parent 3bd5839
commit 749af84
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/cellfinder/core/train/train_yml.py b/cellfinder/core/train/train_yml.py
@@ -338,7 +338,10 @@ def run(
 
     ensure_directory_exists(output_dir)
     model_weights = prep_model_weights(
-        install_path, model_weights, model, n_free_cpus
+        model_weights=model_weights,
+        install_path=install_path,
+        model_name=model,
+        n_free_cpus=n_free_cpus,
     )
 
     yaml_contents = parse_yaml(yaml_file)