fix linting

ealerskans · Feb 10, 2025 · 51e0886 · 51e0886
1 parent 09c500f
commit 51e0886
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/neural_lam/custom_loggers.py b/neural_lam/custom_loggers.py
@@ -1,6 +1,6 @@
 # Standard library
-import sys
 import os
+import sys
 
 # Third-party
 import mlflow
@@ -51,12 +51,12 @@ def log_image(self, key, images, step=None):
         step: Union[int, None]
             Step to log the image under. If None, logs under the key directly
         """
+        # Standard library
+        import os
+
         # Third-party
         from botocore.exceptions import NoCredentialsError
         from PIL import Image
-        import tempfile
-        import pathlib
-        import os
 
         if step is not None:
             key = f"{key}_{step}"

diff --git a/neural_lam/train_model.py b/neural_lam/train_model.py
@@ -23,11 +23,13 @@
     "hi_lam_parallel": HiLAMParallel,
 }
 
+
 def none_or_str(value):
     if value == "None":
         return None
     return value
 
+
 @logger.catch
 def main(input_args=None):
     """Main function for training and evaluating models."""