[tests] clarify RuntimeError in distributed tests(#4452)

microsoft · Jul 7, 2021 · 7f9959f · 7f9959f
1 parent 90342e9
commit 7f9959f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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