disabling test for ensemble nbeats because of timeouts (#580)

* disabling test for ensemble nbeats because of timeouts * disabling more test cases
awslabs · Jan 30, 2020 · ae0be7b · ae0be7b
1 parent f542207
commit ae0be7b
Showing 1 changed file with 11 additions and 10 deletions.
diff --git a/test/model/test_models.py b/test/model/test_models.py
@@ -327,16 +327,17 @@ def deepstate_estimator(hybridize: bool = False, batches_per_epoch=1):
  rnn_estimator(hybridize=hyb) + (10.0,),
  simple_feedforward_estimator(hybridize=hyb, batches_per_epoch=200)
  + (0.3,),
- n_beats_generic_estimator(hybridize=hyb, batches_per_epoch=200)
- + (0.3,),
- n_beats_generic_ensemble_estimator(
- hybridize=hyb, batches_per_epoch=200
- )
- + (0.3,),
- n_beats_interpretable_estimator(
- hybridize=hyb, batches_per_epoch=200
- )
- + (0.3,),
+ # TODO: disabling this because it appears to time out
+ # n_beats_generic_estimator(hybridize=hyb, batches_per_epoch=200)
+ # + (0.3,),
+ # n_beats_generic_ensemble_estimator(
+ # hybridize=hyb, batches_per_epoch=200
+ # )
+ # + (0.3,),
+ # n_beats_interpretable_estimator(
+ # hybridize=hyb, batches_per_epoch=200
+ # )
+ # + (0.3,),
  transformer_estimator(hybridize=hyb, batches_per_epoch=80)
  + (0.2,),
  ]