From 7da0028f3f943719114b2f7b20d7bd19917427d9 Mon Sep 17 00:00:00 2001 From: Saaketh Date: Wed, 5 Jun 2024 17:21:45 -0700 Subject: [PATCH] type --- composer/core/evaluator.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/composer/core/evaluator.py b/composer/core/evaluator.py index 6bed385f77..767131bc35 100644 --- a/composer/core/evaluator.py +++ b/composer/core/evaluator.py @@ -179,7 +179,7 @@ def _get_initial_device_eval_microbatch_size( ), ) from e return batch_size - elif isinstance(device_eval_microbatch_size, Union[int, float]): + elif isinstance(device_eval_microbatch_size, (int, float)): return device_eval_microbatch_size else: raise ValueError("device_eval_microbatch_size must be an int or ``'auto'``")