Skip to content

Commit

Permalink
Increase tolerance + nightly path fix (#318)
Browse files Browse the repository at this point in the history
Co-authored-by: Ashwin Vaidya <ashwinitinvaidya@gmail.com>
  • Loading branch information
ashwinvaidya17 and Ashwin Vaidya committed May 17, 2022
1 parent 9b117b0 commit dd89f1e
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 3 deletions.
4 changes: 2 additions & 2 deletions tests/nightly/models/test_model_nightly.py
Original file line number Diff line number Diff line change
Expand Up @@ -91,7 +91,7 @@ def _test_metrics(self, trainer, config, model, datamodule):
if "optimization" in config.keys() and "nncf" in config.optimization.keys() and config.optimization.nncf.apply:
threshold = threshold.nncf
if not (
np.isclose(results["image_AUROC"], threshold["image_AUROC"], rtol=0.02)
np.isclose(results["image_AUROC"], threshold["image_AUROC"], rtol=0.05)
or (results["image_AUROC"] >= threshold["image_AUROC"])
):
raise AssertionError(
Expand All @@ -100,7 +100,7 @@ def _test_metrics(self, trainer, config, model, datamodule):

if config.dataset.task == "segmentation":
if not (
np.isclose(results["pixel_AUROC"], threshold["pixel_AUROC"], rtol=0.02)
np.isclose(results["pixel_AUROC"], threshold["pixel_AUROC"], rtol=0.05)
or (results["pixel_AUROC"] >= threshold["pixel_AUROC"])
):
raise AssertionError(
Expand Down
2 changes: 1 addition & 1 deletion tests/nightly/tools/benchmarking/test_benchmarking.py
Original file line number Diff line number Diff line change
Expand Up @@ -51,7 +51,7 @@ def check_csv(model: str):

def test_benchmarking():
"""Test if benchmarking script produces the required artifacts."""
config_path = "tests/pre_merge/tools/benchmarking/benchmark_params.yaml"
config_path = "tests/nightly/tools/benchmarking/benchmark_params.yaml"
test_config = OmegaConf.load(config_path)
test_config.grid_search.dataset["path"] = [get_dataset_path()]

Expand Down

0 comments on commit dd89f1e

Please sign in to comment.