diff --git a/requirements-benchmark.txt b/requirements-benchmark.txt index dcd1eaaaea..71eac24cc4 100644 --- a/requirements-benchmark.txt +++ b/requirements-benchmark.txt @@ -4,7 +4,7 @@ # Example requirement, can be anything that pip knows # install with `pip install -r requirements.txt`, and make sure that CI does the same tqdm == 4.59.0 -pandas == 1.2.4 -seaborn == 0.11.1 +pandas == 2.2.2 +seaborn == 0.13.2 pytorch-lightning >= 1.3 torchmetrics>=0.7.0, <0.10.1 diff --git a/xformers/profiler/profiler_dcgm.py b/xformers/profiler/profiler_dcgm.py index 12f2d6cf6d..389cadd2ca 100644 --- a/xformers/profiler/profiler_dcgm.py +++ b/xformers/profiler/profiler_dcgm.py @@ -16,10 +16,6 @@ DCGM_PROFILER_AVAILABLE = True except ModuleNotFoundError: - logger.warning( - f"Unable to find python bindings at {DCGM_PYTHON_PATH}. " - "No data will be captured." - ) class DCGMProfiler: # type: ignore """The dummy DCGM Profiler.""" @@ -34,7 +30,10 @@ def __init__( pass def __enter__(self) -> None: - pass + logger.warning( + f"Unable to find python bindings at {DCGM_PYTHON_PATH}. " + "No data will be captured." + ) def __exit__(self, exc_type, exc_val, exc_tb) -> None: pass