From fcd907c09914f7995cfcec82116696afa00f0276 Mon Sep 17 00:00:00 2001
From: brianreicher <reicher.b@northeastern.edu>
Date: Fri, 15 Dec 2023 11:34:40 -0500
Subject: [PATCH] Fix imports segmentation

---
 src/autoseg/postprocess/segment_mws.py | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/src/autoseg/postprocess/segment_mws.py b/src/autoseg/postprocess/segment_mws.py
index 6efbc7a..7b94f61 100644
--- a/src/autoseg/postprocess/segment_mws.py
+++ b/src/autoseg/postprocess/segment_mws.py
@@ -1,13 +1,11 @@
 # UTILIZES RUSTY_MWS, HGLOM FOR ALL SEGMENTATION, ORIGINALLY WRITTEN BY BRIAN REICHER (2023)
-import torch
 import rusty_mws
 import hglom
 
 from ..utils import neighborhood
 from ..predict.network_predictions import predict_task
-from ..networks.FLibUNet import setup_unet
+from ..networks import *
 from ..models import *
-from ..losses import *
 
 
 def _setup_model(network=setup_unet(), model=MTLSDModel):