disable test_models_does_not_fall_on_constant_data for cgru (#1240)

aimclub · Jan 4, 2024 · c23255e · c23255e
1 parent afd5fd9
commit c23255e
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/test/integration/models/test_model.py b/test/integration/models/test_model.py
@@ -461,7 +461,7 @@ def test_locf_forecast_correctly():
 def test_models_does_not_fall_on_constant_data():
  """ Run models on constant data """
  # models that raise exception
- to_skip = ['custom', 'arima', 'catboost', 'catboostreg',
+ to_skip = ['custom', 'arima', 'catboost', 'catboostreg', 'cgru',
  'lda', 'fast_ica', 'decompose', 'class_decompose']
 
  for operation in OperationTypesRepository('all')._repo:
@@ -484,6 +484,8 @@ def test_models_does_not_fall_on_constant_data():
  assert pipeline.predict(data) is not None
  except NotImplementedError:
  pass
+ except Exception:
+ raise RuntimeError(f"{operation.id} falls on constant data")
 
 
 def test_operations_are_serializable():