From 749af84de1c9c359919d192e8fa1a21d1c55eed9 Mon Sep 17 00:00:00 2001
From: Igor Tatarnikov <61896994+IgorTatarnikov@users.noreply.github.com>
Date: Fri, 16 Feb 2024 11:21:24 +0000
Subject: [PATCH] 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
---
 cellfinder/core/train/train_yml.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/cellfinder/core/train/train_yml.py b/cellfinder/core/train/train_yml.py
index 63e97e00..fbb59968 100644
--- 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)