added ruff lint and black format hooks

Author: SamoraHunter

.githooks/pre-push.sh

@@ -0,0 +1,43 @@
+#!/bin/bash
+# .githooks/pre-push
+
+# Navigate to the repository root (relative to the hook location)
+cd "$(dirname "$0")/.."
+
+# Activate the virtual environment (if needed)
+VENV_PATH=$(find . -type d -name "ml_grid_env")
+if [ -n "$VENV_PATH" ]; then
+  source "$VENV_PATH/bin/activate"
+else
+  echo "Virtual environment not found. Exiting."
+  exit 1
+fi
+
+# Run Ruff linter
+echo "Running Ruff linter..."
+ruff check .
+if [ $? -ne 0 ]; then
+    echo "Ruff linting failed. Please fix the issues before pushing."
+    exit 1
+fi
+
+# Run Black formatter check
+echo "Running Black formatter check..."
+black --check .
+if [ $? -ne 0 ]; then
+    echo "Black formatting check failed. Please run 'black .' to format your code before pushing."
+    exit 1
+fi
+
+# Run the same test command as in the GitHub Actions workflow
+echo "Running tests before pushing..."
+pytest notebooks/test_notebook.py
+
+# Check the exit status of the test command
+if [ $? -ne 0 ]; then
+    echo "Tests failed. Push aborted."
+    exit 1
+fi
+
+echo "All checks passed. Proceeding with push."
+exit 0

config_hyperopt.yml

@@ -31,16 +31,16 @@ data:
 
 # Models to be used during the hyperopt search
 models:
-  LogisticRegression_class: true
-  knn_classifiers_class: true
-  quadratic_discriminant_analysis_class: true
-  SVC_class: true
-  XGB_class_class: true
-  mlp_classifier_class: true
-  RandomForestClassifier_class: true
-  GradientBoostingClassifier_class: true
-  CatBoost_class: true
-  GaussianNB_class: true
+  LogisticRegressionClass: true
+  KNeighborsClassifierClass: true
+  QuadraticDiscriminantAnalysisClass: true
+  SVCClass: true
+  XGBClassifierClass: true
+  MLPClassifierClass: true
+  RandomForestClassifierClass: true
+  GradientBoostingClassifierClass: true
+  CatBoostClassifierClass: true
+  GaussianNBClassifierClass: true
   H2O_class: false # H2O AutoML
   H2O_GBM_class: true
   H2O_DRF_class: true
@@ -51,10 +51,10 @@ models:
   H2O_XGBoost_class: true
   H2O_StackedEnsemble_class: false
   H2O_GAM_class: true
-  LightGBMClassifierWrapper: true
-  adaboost_class: true
-  NeuralNetworkClassifier_class: true
-  TabTransformer_class: false
+  LightGBMClassifierWrapper: true # noqa
+  AdaBoostClassifierClass: true
+  NeuralNetworkClassifier_class: true # noqa
+  TabTransformerClass: false
   # Set the following to true if a GPU is available and configured
   kerasClassifier_class: false
   knn__gpu_wrapper_class: false

pyproject.toml

@@ -53,6 +53,8 @@ dependencies = [
 [project.optional-dependencies]
 # Dependencies for running tests, including notebook tests
 test = [
+    "black",
+    "ruff",
     "pytest",
     "nbmake",
     "scikit-optimize",
@@ -96,4 +98,29 @@ docs = [
 
 
 [tool.setuptools.packages.find]
-include = ["ml_grid"]
+include = ["ml_grid"]
+
+[tool.black]
+line-length = 88
+target-version = ['py310']
+
+[tool.ruff]
+# Set the maximum line length to match Black.
+line-length = 88
+
+# Enable the Pyflakes (F) and pycodestyle (E, W) rules by default.
+# I for isort for import sorting.
+select = ["E", "F", "W", "I"]
+
+# Exclude a variety of commonly ignored directories.
+exclude = [
+    ".git",
+    ".ruff_cache",
+    ".venv",
+    "__pypackages__",
+    "build",
+    "dist",
+    "ml_grid_env",
+    "ml_grid_ts_env",
+    "venv",
+]