updated imports

mlcommons · Jul 28, 2023 · d98472c · d98472c
1 parent 130c99d
commit d98472c
Showing 1 changed file with 6 additions and 8 deletions.
diff --git 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.