Skip to content

Commit

Permalink
Fix function
Browse files Browse the repository at this point in the history
  • Loading branch information
eacharles committed Jul 25, 2024
1 parent 782c7b2 commit 1a5adf3
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/rail/interfaces/tool_factory.py
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand Down

0 comments on commit 1a5adf3

Please sign in to comment.