Update logger.py

logger.py

@@ -2,9 +2,9 @@ import numpy as np
 import torch
 import torch.nn.functional as F
 import imageio
-
-import os
 from skimage.draw import disk
+import os
+
 
 import matplotlib.pyplot as plt
 import collections
@@ -208,4 +208,4 @@ class Visualizer:
 
         image = self.create_image_grid(*images)
         image = (255 * image).astype(np.uint8)
-        return image 
+        return image