Skip to content

Commit 83d96b7

Browse files
committed
Fix tests
1 parent 90c5f66 commit 83d96b7

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

tests/algorithms/test_algorithm_registry.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010
ProgressiveResizingHparams, RandAugmentHparams, SAMHparams, ScaleScheduleHparams,
1111
SelectiveBackpropHparams, SeqLengthWarmupHparams, SqueezeExciteHparams,
1212
StochasticDepthHparams, SWAHparams, algorithm_registry)
13-
from composer.algorithms.hparams import AGCHparams
13+
from composer.algorithms.hparams import AGCHparams, EMAHparams
1414
from composer.core.algorithm import Algorithm
1515

1616
default_required_fields = {
@@ -40,6 +40,7 @@
4040
"p_col": 0.15,
4141
"batch": True,
4242
},
43+
EMAHparams: {},
4344
FactorizeHparams: {
4445
"min_channels": 16,
4546
"latent_channels": 0.5,

0 commit comments

Comments
 (0)