Revert "🚚 Refactor Benchmarking Script (#1216)"

This reverts commit 784767f.
openvinotoolkit · Aug 7, 2023 · d4eaee3 · d4eaee3
1 parent 5a46d03
commit d4eaee3
Show file tree

Hide file tree

Showing 7 changed files with 289 additions and 425 deletions.
diff --git a/pyproject.toml b/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

diff --git a/requirements/loggers.txt b/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
diff --git a/src/anomalib/utils/sweep/__init__.py b/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",
 ]
diff --git a/src/anomalib/utils/sweep/utils.py b/src/anomalib/utils/sweep/utils.py