From 7bc0865f02377059a032a9d7320f3e9902462af5 Mon Sep 17 00:00:00 2001 From: Milo Cress Date: Mon, 17 Jun 2024 18:05:35 +0000 Subject: [PATCH] remove intenum --- composer/algorithms/utils/augmentation_primitives.py | 1 - 1 file changed, 1 deletion(-) diff --git a/composer/algorithms/utils/augmentation_primitives.py b/composer/algorithms/utils/augmentation_primitives.py index dbee50a936..d3ac807c02 100644 --- a/composer/algorithms/utils/augmentation_primitives.py +++ b/composer/algorithms/utils/augmentation_primitives.py @@ -26,7 +26,6 @@ implementations of enhancing color, contrast, brightness, and sharpness. * ``augmentation_sets["all"]`` contains all augmentations. """ -from enum import IntEnum from typing import Callable import numpy as np