From d98472ca4d90ea7e4f491895218a875acb754943 Mon Sep 17 00:00:00 2001
From: Geeks-sid <sid.cre8er@gmail.com>
Date: Fri, 28 Jul 2023 12:48:30 -0400
Subject: [PATCH] updated imports

---
 GANDLF/utils/modelio.py | 14 ++++++--------
 1 file changed, 6 insertions(+), 8 deletions(-)

diff --git a/GANDLF/utils/modelio.py b/GANDLF/utils/modelio.py
index 1111255a2..0deead23f 100644
--- a/GANDLF/utils/modelio.py
+++ b/GANDLF/utils/modelio.py
@@ -1,9 +1,12 @@
-import os, hashlib, pkg_resources, subprocess
+import hashlib
+import os
+import subprocess
+from typing import Any, Dict
+
 import torch
 
-from .generic import get_unique_timestamp
 from ..version import __version__
-from typing import Dict, Any
+from .generic import get_unique_timestamp
 
 # these are the base keys for the model dictionary to save
 model_dict_full = {
@@ -27,11 +30,6 @@
 initial_model_path_end = "_initial.pth.tar"
 
 
-import os
-import torch
-import subprocess
-
-
 def optimize_and_save_model(model, params, path, onnx_export=True):
     """
     Perform post-training optimization and save it to a file.