Skip to content

Commit

Permalink
type (#3368)
Browse files Browse the repository at this point in the history
  • Loading branch information
snarayan21 authored Jun 6, 2024
1 parent 621a3ff commit 4b46f64
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion composer/core/evaluator.py
Original file line number Diff line number Diff line change
@@ -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'``")

0 comments on commit 4b46f64

Please sign in to comment.