cleanup

evaluation/eval_lr.py

@@ -98,7 +98,7 @@ class Evaluator:
                 # Write the header
                 if row == 0:
                     metricsheet.write(1, 0, 'Average')
-                    metricsheet.write(1, 1, f'=AVERAGE(C2:ZZ2)')
+                    metricsheet.write(1, 1, '=AVERAGE(C2:ZZ2)')
                     for col in range(len(metric)):
                         metricsheet.write(0, col + 2, col)
                         colname = xlsxwriter.utility.xl_col_to_name(col + 2)

inference_matanyone2.py

@@ -8,7 +8,7 @@ from PIL import Image
 import torch
 import torch.nn.functional as F
 
-from hugging_face.tools.download_util import load_file_from_url
+from matanyone2.utils.download_util import load_file_from_url
 from matanyone2.utils.inference_utils import gen_dilate, gen_erosion, read_frame_from_videos
 
 from matanyone2.inference.inference_core import InferenceCore

matanyone2/utils/device.py

@@ -1,3 +1,4 @@
+import contextlib
 import torch
 import functools
 
@@ -22,7 +23,6 @@ def safe_autocast_decorator(enabled=True):
         return wrapper
     return decorator
 
-import contextlib
 @contextlib.contextmanager
 def safe_autocast(enabled=True):
     device = get_default_device()