pre-commit fixes

mllam · Oct 30, 2024 · 0eeca64 · 0eeca64
1 parent 39b90a1
commit 0eeca64
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/mllam_data_prep/cli.py b/mllam_data_prep/cli.py
@@ -3,9 +3,10 @@
 """
 import argparse
 import os
-from loguru import logger
 from pathlib import Path
 
+from loguru import logger
+
 from .create_dataset import create_dataset_zarr
 
 # Attempt to import psutil and dask.distributed modules

diff --git a/tests/test_distributed.py b/tests/test_distributed.py
@@ -20,7 +20,7 @@ def distributed():
         importlib.import_module("dask.distributed")
 
         return True
-    except ModuleNotFoundError, ImportError:
+    except (ModuleNotFoundError, ImportError):
         return False