cleaning up

utils.py

@@ -4,7 +4,6 @@ import deepinv as dinv
 
 from models.unext_wip import UNeXt
 from models.unrolled_dpir import get_unrolled_architecture
-from models.PDNet import get_PDNet_architecture
 from physics.multiscale import Pad
 
 
@@ -111,10 +110,7 @@ def get_model(
     """
     model_name = model_name.lower()
 
-    if model_name == "pdnet":
-        return get_PDNet_architecture(in_channels=in_channels, out_channels=in_channels, device=device)
-
-    elif model_name == "unext_emb_physics_config_c":
+    if model_name == "unext_emb_physics_config_c":
         n_chan = [1, 2, 3] # 6 for old head grayscale, complex and color = 1 + 2 + 3
         residual = True if "residual" in model_name else False
         nc = [nc_base * 2**i for i in range(4)]
@@ -149,4 +145,4 @@ def get_model(
         return ArtifactRemoval(model, pinv=False, device=device)
 
     else:
-        raise ValueError(f"Model {model_name} is not supported.")
+        raise ValueError(f"Model {model_name} is not supported.")