Clean up imports

src/StormViewer/gui.py

@@ -1,5 +1,5 @@
 from PyQt6 import QtCore
-from PyQt6.QtWidgets import QApplication, QWidget, QGridLayout, QFileDialog, QMainWindow
+from PyQt6.QtWidgets import QApplication, QWidget, QGridLayout, QFileDialog
 from PyQt6.QtCore import QObject, QThreadPool, QRunnable, Qt
 from PyQt6.QtGui import QIcon
 import os

src/StormViewer/models.py

@@ -2,8 +2,6 @@
 from matplotlib import pyplot as plt
 import seaborn as sns
 from tempfile import NamedTemporaryFile
-import os
-from typing import Union
 
 
 class POLine:

test/test.py

@@ -1,7 +1,7 @@
 import unittest
 import os
 import pandas as pd
-from tuflow_ensemble import te, logger
+from StormViewer import te, logger
 import pathlib
 
 wd = pathlib.Path(__file__).parent.resolve()