checking new syntax

mlcommons · Nov 27, 2023 · fe6e1ba · fe6e1ba
1 parent 99e8eee
commit fe6e1ba
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/GANDLF/compute/forward_pass.py b/GANDLF/compute/forward_pass.py
@@ -150,7 +150,6 @@ def validate_network(
         for key in params["channel_keys"]:
             subject_dict[key] = torchio.ScalarImage(
                 path=subject[key]["path"],
-                type=subject[key]["type"],
                 tensor=subject[key]["data"].squeeze(0),
                 affine=subject[key]["affine"].squeeze(0),
             )
@@ -320,7 +319,6 @@ def validate_network(
                 output_prediction = output_prediction.unsqueeze(0)
                 if params["save_output"]:
                     img_for_metadata = torchio.ScalarImage(
-                        type=subject["1"]["type"],
                         tensor=subject["1"]["data"].squeeze(0),
                         affine=subject["1"]["affine"].squeeze(0),
                     ).as_sitk()