Updates by running pre-commit hooks

Author: miguelgondu

src/hdbo_benchmark/experiments/training_vae_on_zinc250k/train_vae.py

@@ -1,9 +1,9 @@
 import click
 import numpy as np
 import torch
+import wandb
 
 import hdbo_benchmark
-import wandb
 from hdbo_benchmark.generative_models.vae_selfies import VAESelfies
 from hdbo_benchmark.utils.constants import DEVICE, WANDB_ENTITY
 from hdbo_benchmark.utils.data.zinc_250k import load_zinc_250k_dataloaders

src/hdbo_benchmark/experiments/training_vae_on_zinc250k/train_vae_rnn.py

@@ -1,9 +1,9 @@
 import click
 import numpy as np
 import torch
+import wandb
 
 import hdbo_benchmark
-import wandb
 from hdbo_benchmark.generative_models.vae_rnn_selfies import VAERNNSelfies
 from hdbo_benchmark.utils.constants import DEVICE, WANDB_ENTITY
 from hdbo_benchmark.utils.data.zinc_250k import load_zinc_250k_dataloaders

src/hdbo_benchmark/results/benchmark_on_pmo/create_table.py

@@ -26,8 +26,8 @@
 """
 
 import pandas as pd
-
 import wandb
+
 from hdbo_benchmark.utils.constants import ROOT_DIR
 from hdbo_benchmark.utils.results.download_from_wandb import get_all_runs_for_experiment
 

src/hdbo_benchmark/results/benchmark_on_pmo/visualize_function_vs_solver_heatmap.py

@@ -1,6 +1,4 @@
-"""
-
-"""
+""" """
 
 import matplotlib
 import matplotlib.pyplot as plt

src/hdbo_benchmark/results/download_tables.py

@@ -26,8 +26,8 @@
 """
 
 import pandas as pd
-
 import wandb
+
 from hdbo_benchmark.utils.constants import ROOT_DIR
 from hdbo_benchmark.utils.results.download_from_wandb import (
     get_all_runs_for_function_names,

src/hdbo_benchmark/utils/ax/ax_solver.py

@@ -54,7 +54,7 @@ def __init__(
         )
 
         def evaluate(
-            parametrization: dict[str, float]
+            parametrization: dict[str, float],
         ) -> dict[str, tuple[float, float]]:
             x = np.array([[parametrization[f"x{i}"] for i in range(x0.shape[1])]])
             y = black_box(x)

src/hdbo_benchmark/utils/data/experiments/loading_results.py

@@ -5,8 +5,8 @@
 from typing import Any, Dict, List
 
 import pandas as pd
-
 import wandb
+
 from hdbo_benchmark.utils.constants import WANDB_ENTITY
 
 ROOT_DIR = Path(__file__).parent.parent.parent.parent.parent.parent.resolve()

src/hdbo_benchmark/utils/experiments/load_generative_models.py

@@ -15,9 +15,9 @@ def load_generative_model_and_bounds(
     problem: Problem,
 ) -> tuple[VAE | LitAutoEncoder, tuple[float, float]]:
     match function_name:
-        case function_name if function_name in PMOBenchmark(
-            string_representation="SELFIES"
-        ).problem_names:
+        case function_name if (
+            function_name in PMOBenchmark(string_representation="SELFIES").problem_names
+        ):
             experiment_name = "benchmark_on_pmo"
             latent_space_bounds = (-10.0, 10.0)
             vae = VAEFactory().create(

src/hdbo_benchmark/utils/experiments/load_problems.py

@@ -147,9 +147,9 @@ def _load_ehrlich_holo(size: Literal["tiny", "small", "large"]) -> Problem:
 
 def _load_problem(function_name: str) -> Problem:
     match function_name:
-        case function_name if function_name in PMOBenchmark(
-            string_representation="SELFIES"
-        ).problem_names:
+        case function_name if (
+            function_name in PMOBenchmark(string_representation="SELFIES").problem_names
+        ):
             return _load_pmo_problem(function_name)
         case "rfp_rasp":
             return _load_rasp()

src/hdbo_benchmark/utils/logging/back_up.py

@@ -8,8 +8,8 @@
 from time import time
 
 import pandas as pd
-
 import wandb
+
 from hdbo_benchmark.utils.constants import ROOT_DIR, WANDB_ENTITY, WANDB_PROJECT
 from hdbo_benchmark.utils.data.experiments.loading_results import _load_run_as_row