4966 fix deepedit get clicks (#4969)

Fixes #4966

### Types of changes
<!--- Put an `x` in all the boxes that apply, and remove the not applicable items -->
- [x] Non-breaking change (fix or new feature that would not break existing functionality).
- [ ] Breaking change (fix or new feature that would cause existing functionality to change).
- [ ] New tests added to cover the changes.
- [x] Integration tests passed locally by running `./runtests.sh -f -u --net --coverage`.
- [x] Quick tests passed locally by running `./runtests.sh --quick --unittests  --disttests`.
- [ ] In-line docstrings updated.
- [ ] Documentation updated, tested `make html` command in the `docs/` folder.

Signed-off-by: wojiazaiyugang <[email protected]>

Author: wojiazaiyugang

monai/apps/deepedit/transforms.py

@@ -673,7 +673,7 @@ def __call__(self, data):
         # Creating guidance for all clicks
         all_guidances = {}
         for key_label in self.label_names.keys():
-            clicks = d.get(key_label, [])
+            clicks = d[self.guidance].get(key_label, [])
             clicks = list(np.array(clicks).astype(int))
             all_guidances[key_label] = self._apply(clicks, factor)
         d[self.guidance] = all_guidances

tests/test_deepedit_transforms.py

@@ -79,7 +79,7 @@
     "current_label": "spleen",
     "probability": 1.0,
     "label_names": LABEL_NAMES,
-    "spleen": [[0, 4, 3], [0, 0, 3], [0, 1, 3]],
+    "guidance": {"spleen": [[0, 4, 3], [0, 0, 3], [0, 1, 3]]},
     "sids": {"spleen": []},
     "pred": PRED,
 }
@@ -140,7 +140,6 @@
 
 DATA_11 = {"image": IMAGE, "label": LABEL, "label_names": LABEL_NAMES, "pred": PRED}
 
-
 ADD_GUIDANCE_FROM_POINTS_TEST_CASE = [
     {"ref_image": "image", "guidance": "guidance", "label_names": LABEL_NAMES},  # arguments
     DATA_4,  # input_data