List Comprehension for ITC (#2906)

Project-MONAI · Sep 8, 2021 · e24f1b1 · e24f1b1
1 parent b770793
commit e24f1b1
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 5 deletions.
diff --git a/monai/apps/pathology/utils.py b/monai/apps/pathology/utils.py
@@ -51,10 +51,7 @@ def compute_isolated_tumor_cells(tumor_mask: np.ndarray, threshold: float) -> Li
     """
     max_label = np.amax(tumor_mask)
     properties = measure.regionprops(tumor_mask, coordinates="rc")
-    itc_list = []
-    for i in range(max_label):  # type: ignore
-        if properties[i].major_axis_length < threshold:
-            itc_list.append(i + 1)
+    itc_list = [i + 1 for i in range(max_label) if properties[i].major_axis_length < threshold]
 
     return itc_list
 

diff --git a/monai/handlers/garbage_collector.py b/monai/handlers/garbage_collector.py
@@ -69,7 +69,7 @@ def __call__(self, engine: Engine) -> None:
         """
         # get count before garbage collection
         pre_count = gc.get_count()
-        # fits call to garbage collector
+        # first call to garbage collector
         gc.collect()
         # second call to garbage collector
         unreachable = gc.collect()