From c485a771c7e73e6ce2136bd8283ea6ffccde9e40 Mon Sep 17 00:00:00 2001 From: Greg Starr Date: Wed, 24 May 2023 13:42:29 -0400 Subject: [PATCH] small linting fix --- .github/workflows/python-package.yml | 2 +- volumentations/augmentations/functional.py | 2 ++ 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/.github/workflows/python-package.yml b/.github/workflows/python-package.yml index 47ba5ea..ccea277 100644 --- a/.github/workflows/python-package.yml +++ b/.github/workflows/python-package.yml @@ -31,7 +31,7 @@ jobs: poetry install --with dev - name: Lint with flake8 run: | - poetry run flake8 . --count --select=E9,F63,F7,F82 --show-source --statistics + poetry run flake8 ./volumentations --count --select=E9,F63,F7,F82 --show-source --statistics - name: Test with pytest run: | pytest diff --git a/volumentations/augmentations/functional.py b/volumentations/augmentations/functional.py index d04c8e3..9170ba3 100644 --- a/volumentations/augmentations/functional.py +++ b/volumentations/augmentations/functional.py @@ -43,6 +43,8 @@ from scipy.ndimage import gaussian_filter from scipy.ndimage import map_coordinates from warnings import warn +from itertools import product + MAX_VALUES_BY_DTYPE = { np.dtype("uint8"): 255,