Revert "🚚 Refactor Benchmarking Script"

pyproject.toml

@@ -58,9 +58,6 @@ target-version = "py38"
 # Allow imports relative to the "src" and "tests" directories.
 src = ["src", "tests"]
 
-[tool.ruff.isort]
-known-third-party = ["wandb", "comet_ml"]
-
 [tool.ruff.mccabe]
 # Unlike Flake8, default to a complexity level of 10.
 max-complexity = 10

requirements/loggers.txt

@@ -1,6 +1,6 @@
 comet-ml>=3.31.7
 gradio>=2.9.4
 tensorboard
-wandb>=0.13.0
+wandb==0.12.17
 GitPython
 ipykernel

src/anomalib/utils/sweep/__init__.py

@@ -5,16 +5,12 @@
 
 from .config import flatten_sweep_params, get_run_config, set_in_nested_config
 from .helpers import get_openvino_throughput, get_sweep_callbacks, get_torch_throughput
-from .utils import Status, exception_wrapper, redirect_output
 
 __all__ = [
-    "exception_wrapper",
-    "flatten_sweep_params",
     "get_run_config",
+    "set_in_nested_config",
     "get_sweep_callbacks",
     "get_openvino_throughput",
     "get_torch_throughput",
-    "set_in_nested_config",
-    "redirect_output",
-    "Status",
+    "flatten_sweep_params",
 ]