Merge branch 'fix_flow_tags' of ssh://github.com/openml/openml-python into fix_flow_tags

Author: mfeurer

tests/test_runs/test_run_functions.py

@@ -63,9 +63,13 @@ def _check_serialized_optimized_run(self, run_id):
         # downloads the best model based on the optimization trace
         # suboptimal (slow), and not guaranteed to work if evaluation
         # engine is behind. TODO: mock this? We have the arff already on the server
-        self._wait_for_processed_run(run_id, 80)
-        model_prime = openml.runs.initialize_model_from_trace(run_id, 0, 0)
-
+        self._wait_for_processed_run(run_id, 200)
+        try:
+            model_prime = openml.runs.initialize_model_from_trace(run_id, 0, 0)
+        except openml.exceptions.OpenMLServerException as e:
+            e.additional += '; run_id: ' + run_id
+            raise e
+        
         run_prime = openml.runs.run_model_on_task(task, model_prime,
                                                   avoid_duplicate_runs=False,
                                                   seed=1)
@@ -357,7 +361,7 @@ def test_get_run_trace(self):
             # in case the run did not exists yet
             run = openml.runs.run_model_on_task(task, clf, avoid_duplicate_runs=True)
             run = run.publish()
-            self._wait_for_processed_run(run.run_id, 80)
+            self._wait_for_processed_run(run.run_id, 200)
             run_id = run.run_id
         except openml.exceptions.PyOpenMLError:
             # run was already