Rm unused global, nonlocal

Author: ericup

celldetection/util/timer.py

@@ -69,7 +69,6 @@ def start_timer(name, cuda=True, collect=True):
     Returns:
 
     """
-    global TIMINGS
     if collect:
         gc.collect()
     if cuda:
@@ -80,7 +79,6 @@ def start_timer(name, cuda=True, collect=True):
 
 
 def stop_timer(name, cuda=True, verbose=True):
-    global TIMINGS
     if cuda:
         torch.cuda.synchronize()
     delta = time() - TIMINGS[name]

celldetection_scripts/cpn_inference.py

@@ -175,7 +175,7 @@ def oom_safe_concat_results_flat_(coll, new, target_device=None, fallback_device
 
         # OOM safe concatenation
         def on_oom():
-            nonlocal target_device, coll
+            nonlocal target_device
             warn(f'Not enough memory. Moving data to {fallback_device} in order to continue.')
             target_device = fallback_device
             for k__, v__ in coll.items():