Skip to content

Commit 5704c29

Browse files
committed
Fix arg name
1 parent bdf157e commit 5704c29

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

models/bpreg/utils/BPRRunner.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -17,9 +17,9 @@ class BPRRunner(Module):
1717
@IO.Instance()
1818
@IO.Input('in_data', 'nifti:mod=ct', the='The input NIfTI CT image file.')
1919
@IO.Output('bpreg_json', 'bpreg.json', "json:type=bpreg", bundle='model', the=' The output JSON file to save the extracted features.')
20-
def task(self, instance: Instance, in_data: InstanceData, utility_json: InstanceData) -> None:
20+
def task(self, instance: Instance, in_data: InstanceData, bpreg_json: InstanceData) -> None:
2121
gpu_available = torch.cuda.is_available()
2222
model = InferenceModel("/app/public_inference_model/public_bpr_model/", gpu=gpu_available)
2323
input_path = in_data.abspath
24-
output_path = utility_json.abspath
24+
output_path = bpreg_json.abspath
2525
model.nifti2json(input_path, output_path, stringify_json=False)

0 commit comments

Comments
 (0)