977 fix dynunet task2 type issue (#978)

Fixes #977  .

### Description
This PR adds the cast transform to prevent type issue on dynunet
pipeline for decathlon task 2 dataset.

### Checks
<!--- Put an `x` in all the boxes that apply, and remove the not
applicable items -->
- [ ] Notebook runs automatically `./runner [-p <regex_pattern>]`

Signed-off-by: Yiheng Wang <vennw@nvidia.com>

Author: yiheng-wang-nv

modules/dynunet_pipeline/transforms.py

@@ -67,11 +67,22 @@ def get_task_transforms(mode, task_id, pos_sample_num, neg_sample_num, num_sampl
             RandFlipd(["image", "label"], spatial_axis=[0], prob=0.5),
             RandFlipd(["image", "label"], spatial_axis=[1], prob=0.5),
             RandFlipd(["image", "label"], spatial_axis=[2], prob=0.5),
+            CastToTyped(keys=["image", "label"], dtype=(np.float32, np.uint8)),
+            EnsureTyped(keys=["image", "label"]),
+        ]
+    elif mode == "validation":
+        other_transforms = [
+            CastToTyped(keys=["image", "label"], dtype=(np.float32, np.uint8)),
+            EnsureTyped(keys=["image", "label"]),
         ]
-
-        return Compose(load_transforms + sample_transforms + other_transforms)
     else:
-        return Compose(load_transforms + sample_transforms)
+        other_transforms = [
+            CastToTyped(keys=["image"], dtype=(np.float32)),
+            EnsureTyped(keys=["image"]),
+        ]
+
+    all_transforms = load_transforms + sample_transforms + other_transforms
+    return Compose(all_transforms)
 
 def resample_image(image, shape, anisotrophy_flag):
     resized_channels = []