Adding max_retries to config default

openml · PGijsbers · Feb 11, 2021 · Oct 29, 2020 · Oct 29, 2020 · Oct 29, 2020
commit 8e8ea2e5cd611112ce7ece5fd6d421f45107ffea
diff --git a/openml/_api_calls.py b/openml/_api_calls.py
@@ -175,7 +175,7 @@ def _send_request(
     request_method, url, data, files=None,
 ):
     n_retries = config.connection_n_retries
-    max_retries = 10
+    max_retries = config.max_retries
     retry_counter = 0
     response = None
     with requests.Session() as session:
@@ -199,13 +199,13 @@ def _send_request(
                 OpenMLServerException,
             ) as e:
                 if isinstance(e, OpenMLServerException):
-                    if e.code != 107:
-                        # 107 is a database connection error - only then do retries
-                        raise e
-                    else:
+                    if e.code in [107, 500]:
+                        # 107: database connection error
+                        # 500: internal server error
                         wait_time = 0.3
-                        # increase retries if database connection error
-                        n_retries = min(n_retries + 1, max_retries)
+                        n_retries = min(n_retries + 1, max_retries)  # increase retries
+                    else:
+                        raise
                 else:
                     wait_time = 0.1
                 if retry_counter == n_retries:

diff --git a/openml/config.py b/openml/config.py
@@ -88,6 +88,7 @@ def set_file_log_level(file_output_level: int):
     "cachedir": os.path.expanduser(os.path.join("~", ".openml", "cache")),
     "avoid_duplicate_runs": "True",
     "connection_n_retries": 2,
+    "max_retries": 20,
 }
 
 config_file = os.path.expanduser(os.path.join("~", ".openml", "config"))
@@ -116,6 +117,7 @@ def get_server_base_url() -> str:
 
 # Number of retries if the connection breaks
 connection_n_retries = _defaults["connection_n_retries"]
+max_retries = _defaults["max_retries"]
 
 
 class ConfigurationForExamples:
@@ -183,6 +185,7 @@ def _setup():
     global cache_directory
     global avoid_duplicate_runs
     global connection_n_retries
+    global max_retries
 
     # read config file, create cache directory
     try:
@@ -207,10 +210,11 @@ def _setup():
 
     avoid_duplicate_runs = config.getboolean("FAKE_SECTION", "avoid_duplicate_runs")
     connection_n_retries = config.get("FAKE_SECTION", "connection_n_retries")
-    if connection_n_retries > 20:
+    max_retries = config.get("FAKE_SECTION", "max_retries")
+    if connection_n_retries > max_retries:
         raise ValueError(
-            "A higher number of retries than 20 is not allowed to keep the "
-            "server load reasonable"
+            "A higher number of retries than {} is not allowed to keep the "
+            "server load reasonable".format(max_retries)
         )
 
 

diff --git a/openml/testing.py b/openml/testing.py
@@ -261,15 +261,6 @@ def check_task_existence(
     Parameter
     ---------
     task_type : openml.tasks.TaskType
-        ID of the task type as detailed `here <https://www.openml.org/search?type=task_type>`_.
-        - Supervised classification: 1
-        - Supervised regression: 2
-        - Learning curve: 3
-        - Supervised data stream classification: 4
-        - Clustering: 5
-        - Machine Learning Challenge: 6
-        - Survival Analysis: 7
-        - Subgroup Discovery: 8
     dataset_id : int
     target_name : str
 

diff --git a/tests/test_runs/test_run_functions.py b/tests/test_runs/test_run_functions.py
@@ -562,7 +562,6 @@ def test_run_and_upload_linear_regression(self):
         if _task_id is not None:
             task_id = _task_id
         else:
-            task_meta_data["task_type"] = TaskType.SUPERVISED_REGRESSION
             new_task = openml.tasks.create_task(**task_meta_data)
             # publishes the new task
             try:
@@ -996,7 +995,6 @@ def test_initialize_model_from_run(self):
         if _task_id is not None:
             task_id = _task_id
         else:
-            task_meta_data["task_type"] = TaskType.SUPERVISED_CLASSIFICATION
             new_task = openml.tasks.create_task(**task_meta_data)
             # publishes the new task
             try:
@@ -1556,7 +1554,6 @@ def test_format_prediction_task_regression(self):
         if _task_id is not None:
             task_id = _task_id
         else:
-            task_meta_data["task_type"] = TaskType.SUPERVISED_REGRESSION
             new_task = openml.tasks.create_task(**task_meta_data)
             # publishes the new task
             try:

diff --git a/tests/test_study/test_study_functions.py b/tests/test_study/test_study_functions.py
@@ -213,9 +213,8 @@ def test_study_attach_illegal(self):
     def test_study_list(self):
         study_list = openml.study.list_studies(status="in_preparation")
         # might fail if server is recently resetted
-        self.assertGreater(len(study_list), 2)
+        self.assertGreaterEqual(len(study_list), 2)
 
     def test_study_list_output_format(self):
         study_list = openml.study.list_studies(status="in_preparation", output_format="dataframe")
         self.assertIsInstance(study_list, pd.DataFrame)
-        self.assertGreater(len(study_list), 2)
diff --git a/tests/test_tasks/test_regression_task.py b/tests/test_tasks/test_regression_task.py
@@ -28,7 +28,6 @@ def setUp(self, n_levels: int = 1):
         if _task_id is not None:
             task_id = _task_id
         else:
-            task_meta_data["task_type"] = TaskType.SUPERVISED_REGRESSION
             new_task = openml.tasks.create_task(**task_meta_data)
             # publishes the new task
             try: