From 7f9959fe1c04b0559c3d1e093f7790565769bf0c Mon Sep 17 00:00:00 2001 From: Nikita Titov Date: Wed, 7 Jul 2021 16:29:53 +0300 Subject: [PATCH] [tests] clarify RuntimeError in distributed tests(#4452) --- tests/distributed/_test_distributed.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/distributed/_test_distributed.py b/tests/distributed/_test_distributed.py index b86d73cc7ce6..9e1dd8e4f5a4 100644 --- a/tests/distributed/_test_distributed.py +++ b/tests/distributed/_test_distributed.py @@ -148,7 +148,7 @@ def predict(self, predict_config: Dict[str, Any] = {}) -> np.ndarray: cmd = [self.executable, f'config={config_path}'] result = subprocess.run(cmd) if result.returncode != 0: - raise RuntimeError + raise RuntimeError('Error in prediction') y_pred = np.loadtxt(str(TESTS_DIR / 'predictions.txt')) return y_pred