From 099bb7041c3d82eb87e1180bbdc13af32a2ee1cb Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Fran=C3=A7ois=20Laurent?= <francois.laurent@posteo.net>
Date: Thu, 21 Nov 2024 18:26:53 +0100
Subject: [PATCH] fix: argument parsing broken for train (3)
---
src/Taggers.jl | 10 +++++-----
1 file changed, 5 insertions(+), 5 deletions(-)
diff --git a/src/Taggers.jl b/src/Taggers.jl
index 09520c1..4844907 100644
--- a/src/Taggers.jl
+++ b/src/Taggers.jl
@@ -227,9 +227,9 @@ function parsekwargs!(args, kwargs)
end
function run(tagger, switch, kwargs)
- kwargs = Dict{AbstractString, Any}(kwargs)
- kwargs["model_instance"] = tagger.model_instance
- kwargs["sandbox"] = tagger.sandbox
+ kwargs = Dict{Symbol, Any}(kwargs)
+ kwargs[:model_instance] = tagger.model_instance
+ kwargs[:sandbox] = tagger.sandbox
args = Any[]
parsekwargs!(args, kwargs)
cmd = tagging_backend_command(tagger)
@@ -245,8 +245,8 @@ end
predict(tagger::Tagger; kwargs...) = run(tagger, "predict", kwargs)
function finetune(tagger::Tagger; original_instance=nothing, kwargs...)
- kwargs = Dict{AbstractString, Any}(kwargs)
- kwargs["original_model_instance"] = original_instance
+ kwargs = Dict{Symbol, Any}(kwargs)
+ kwargs[:original_model_instance] = original_instance
ret = run(tagger, "finetune", kwargs)
@assert isdir(modeldir(tagger))
return ret
--
GitLab