Fixed imports

.pre-commit-config.yaml

@@ -42,11 +42,6 @@ repos:
         language: system
         types: [python]
         pass_filenames: false
-      - id: ruff
-        name: ruff
-        entry: poetry run ruff check
-        language: system
-        types: [python]
       - id: flake8
         name: flake8
         entry: poetry run flake8

setup.cfg

@@ -36,6 +36,8 @@ ignore =
     # Module level import not at top of file
     C901,
     # Too complex
+    F401,
+    # Unused imports
 
 # pep8-naming
 classmethod-decorators =

src/capfinder/ml_libs.py

@@ -1,11 +1,30 @@
-import os
+import logging  # noqa
+import os  # noqa
+import warnings  # noqa
 
-os.environ["KERAS_BACKEND"] = "jax"
+import jax  # noqa
 
-# from keras import Model
+os.environ["KERAS_BACKEND"] = "jax"
 
-import logging
-import warnings
+import keras  # noqa
+from keras import layers  # noqa
+from keras.layers import LSTM  # noqa
+from keras.layers import BatchNormalization  # noqa
+from keras.layers import Conv1D  # noqa
+from keras.layers import Dense  # noqa
+from keras.layers import Dropout  # noqa
+from keras.layers import Input  # noqa
+from keras.layers import MaxPooling1D  # noqa
+from keras.models import Model  # noqa
+from keras.optimizers import Adam  # noqa
+from keras_tuner import (  # noqa
+    BayesianOptimization,
+    Hyperband,
+    HyperModel,
+    HyperParameters,
+    Objective,
+    RandomSearch,
+)
 
 # Disable TensorFlow logging
 os.environ["TF_CPP_MIN_LOG_LEVEL"] = "3"  # FATAL
@@ -18,3 +37,4 @@
 warnings.filterwarnings("ignore", category=UserWarning, message=".*TensorRT.*")
 
 # Now import TensorFlow
+import tensorflow as tf  # noqa

src/capfinder/training.py

@@ -10,15 +10,8 @@
 import pandas as pd
 from comet_ml import Experiment  # Import CometML before keras
 from loguru import logger
-from ml_libs import (
-    BayesianOptimization,
-    Hyperband,
-    Objective,
-    RandomSearch,
-    jax,
-    keras,
-    tf,
-)
+from ml_libs import jax  # noqa
+from ml_libs import BayesianOptimization, Hyperband, Objective, RandomSearch, keras, tf
 from sklearn.metrics import confusion_matrix
 from tqdm import tqdm