From 1a5adf33e2dd879a62a41984f7b3cf1186a1aadb Mon Sep 17 00:00:00 2001 From: Eric Charles Date: Thu, 25 Jul 2024 16:32:16 -0700 Subject: [PATCH] Fix function --- src/rail/interfaces/tool_factory.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/rail/interfaces/tool_factory.py b/src/rail/interfaces/tool_factory.py index 124dac0..86c778b 100644 --- a/src/rail/interfaces/tool_factory.py +++ b/src/rail/interfaces/tool_factory.py @@ -51,7 +51,7 @@ def build_tool_stage( Newly constructed and configured Estimator instance """ stage_class = PipelineStage.get_stage(class_name, module_name) - stage_obj = stage_class.make_stage(name=stage_name, model=model_path, input=data_path, **config_params) + stage_obj = stage_class.make_stage(name=stage_name, input=data_path, **config_params) cls._stage_dict[stage_name] = stage_obj return stage_obj