0.1.0.dev1 release flake8 fix

pytorch_unet/optimize/config.py

@@ -4,4 +4,4 @@
 
 This can be like the task file for hp tuning, you will use config parser and have the configs saved in a different
 config.ini file
-"""
+"""

pytorch_unet/optimize/hyperparameter.py

@@ -1,2 +1,2 @@
 """write the file such that you can call the training from here and add a hyperparameter tuning to the task
-so that that calls this file and this starts hyperparameter tuning"""
+so that that calls this file and this starts hyperparameter tuning"""

pytorch_unet/utils/proto.py

@@ -1,14 +1,14 @@
 import tifffile as tiff
+import torch
 from PIL import Image
 from sklearn.model_selection import train_test_split
 from torch import nn
 from torch.optim import Adam
 from torch.utils.data import DataLoader
 from torch.utils.data.sampler import RandomSampler
-from torchvision.transforms import Resize, ToTensor
+from torchvision.transforms import Resize, ToTensor, Compose
 
 from pytorch_unet.model.u_net import UNet
-from pytorch_unet.processing.augments import *
 from pytorch_unet.processing.load import DataTransformer
 
 torch.set_default_tensor_type('torch.FloatTensor')

pytorch_unet/visualize/logger.py

@@ -52,5 +52,3 @@ def histo_summary(self, tag, values, step, bins=1000):
         summary = tf.Summary(value=[tf.Summary.Value(tag=tag, histo=hist)])
         self.writer.add_summary(summary, step)
         self.writer.flush()
-
-