fix imports (#5)

fix import statements
eeulig · Nov 18, 2024 · 4df6fb8 · 4df6fb8
1 parent 3319dc0
commit 4df6fb8
Show file tree

Hide file tree

Showing 6 changed files with 18 additions and 12 deletions.
diff --git a/ldctbench/methods/bilateral/Trainer.py b/ldctbench/methods/bilateral/Trainer.py
@@ -2,8 +2,9 @@
 
 import torch
 import torch.nn as nn
-import utils
-from methods.base import BaseTrainer
+
+from ldctbench import utils
+from ldctbench.methods.base import BaseTrainer
 
 from .network import Model
 

diff --git a/ldctbench/methods/dugan/Trainer.py b/ldctbench/methods/dugan/Trainer.py
@@ -5,11 +5,12 @@
 import torch.nn as nn
 import torch.nn.functional as F
 import torch.optim as optim
-import utils
-from methods.base import BaseTrainer
 from torch.autograd import Variable
 from tqdm import tqdm
 
+from ldctbench import utils
+from ldctbench.methods.base import BaseTrainer
+
 from .network import Model, UNet
 from .utils import SobelOperator, cutmix, ls_gan, mask_src_tgt, turn_on_spectral_norm
 

diff --git a/ldctbench/methods/qae/Trainer.py b/ldctbench/methods/qae/Trainer.py
@@ -2,8 +2,9 @@
 
 import torch
 import torch.nn as nn
-import utils
-from methods.base import BaseTrainer
+
+from ldctbench import utils
+from ldctbench.methods.base import BaseTrainer
 
 from .network import Model
 

diff --git a/ldctbench/methods/resnet/Trainer.py b/ldctbench/methods/resnet/Trainer.py
@@ -2,8 +2,9 @@
 
 import torch
 import torch.nn as nn
-import utils
-from methods.base import BaseTrainer
+
+from ldctbench import utils
+from ldctbench.methods.base import BaseTrainer
 
 from .network import Model
 

diff --git a/ldctbench/methods/transct/Trainer.py b/ldctbench/methods/transct/Trainer.py
@@ -2,8 +2,9 @@
 
 import torch
 import torch.nn as nn
-import utils
-from methods.base import BaseTrainer
+
+from ldctbench import utils
+from ldctbench.methods.base import BaseTrainer
 
 from .network import Model
 

diff --git a/ldctbench/methods/wganvgg/Trainer.py b/ldctbench/methods/wganvgg/Trainer.py
@@ -4,8 +4,9 @@
 import numpy as np
 import torch
 import torch.nn as nn
-import utils
-from methods.base import BaseTrainer
+
+from ldctbench import utils
+from ldctbench.methods.base import BaseTrainer
 
 from .network import Discriminator, Model