Skip to content

Commit

Permalink
Fixes from comments
Browse files Browse the repository at this point in the history
  • Loading branch information
GalyaZalesskaya committed Sep 27, 2024
1 parent 08a819c commit 46aaf6c
Show file tree
Hide file tree
Showing 3 changed files with 4 additions and 4 deletions.
2 changes: 1 addition & 1 deletion openvino_xai/metrics/adcc.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,7 @@ class ADCC(BaseMetric):
https://github.com/aimagelab/ADCC/
"""

def __init__(self, model, preprocess_fn, postprocess_fn, explainer, device_name="AUTO", **kwargs: Any):
def __init__(self, model, preprocess_fn, postprocess_fn, explainer, device_name="CPU", **kwargs: Any):
super().__init__(
model=model, preprocess_fn=preprocess_fn, postprocess_fn=postprocess_fn, device_name=device_name
)
Expand Down
2 changes: 1 addition & 1 deletion openvino_xai/metrics/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@ def __init__(
model: ov.Model = None,
preprocess_fn: Callable[[np.ndarray], np.ndarray] = IdentityPreprocessFN(),
postprocess_fn: Callable[[np.ndarray], np.ndarray] = None,
device_name: str = "AUTO",
device_name: str = "CPU",
):
# Pass model_predict to class initialization directly?
self.model = model
Expand Down
4 changes: 2 additions & 2 deletions tests/perf/conftest.py
Original file line number Diff line number Diff line change
Expand Up @@ -33,7 +33,7 @@ def pytest_addoption(parser: pytest.Parser):
help="Number of masks for black box methods." "Defaults to 5000.",
)
parser.addoption(
"--dataset-data-root",
"--dataset-root",
action="store",
default="",
help="Path to directory with dataset images.",
Expand Down Expand Up @@ -190,7 +190,7 @@ def fxt_perf_summary(
@pytest.fixture(scope="session")
def fxt_dataset_parameters(request: pytest.FixtureRequest) -> tuple[Path | None, Path | None]:
"""Retrieve dataset parameters for tests."""
data_root = request.config.getoption("--dataset-data-root")
data_root = request.config.getoption("--dataset-root")
ann_path = request.config.getoption("--dataset-ann-path")

if data_root != "":
Expand Down

0 comments on commit 46aaf6c

Please sign in to comment.