From 4c0d787e8b8aed161d03ccbb483c94a868855036 Mon Sep 17 00:00:00 2001 From: Samet Date: Mon, 19 Feb 2024 20:09:48 +0000 Subject: [PATCH] Fix albumentation tests --- src/anomalib/data/utils/transforms.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/anomalib/data/utils/transforms.py b/src/anomalib/data/utils/transforms.py index 930f3710bd..be40c99d99 100644 --- a/src/anomalib/data/utils/transforms.py +++ b/src/anomalib/data/utils/transforms.py @@ -120,7 +120,7 @@ def get_transforms( # load transforms from config file elif isinstance(config, str): logger.info("Reading transforms from Albumentations config file: %s.", config) - transforms = A.load(filepath=config, data_format="yaml") + transforms = A.load(filepath_or_buffer=config, data_format="yaml") elif isinstance(config, A.Compose): logger.info("Transforms loaded from Albumentations Compose object") transforms = config