Flake8. Add thread safety to get_run.

Author: PGijsbers

openml/datasets/functions.py

@@ -23,8 +23,7 @@
 from ..utils import (
     _create_cache_directory,
     _remove_cache_dir_for_id,
-    _create_cache_directory_for_id,
-    _create_lockfiles_dir,
+    _create_cache_directory_for_id
 )
 
 

openml/runs/functions.py

@@ -466,6 +466,7 @@ def get_runs(run_ids):
     return runs
 
 
+@openml.utils.thread_safe_if_oslo_installed
 def get_run(run_id):
     """Gets run corresponding to run_id.
 

openml/utils.py

@@ -294,7 +294,7 @@ def _remove_cache_dir_for_id(key, cache_dir):
 def thread_safe_if_oslo_installed(func):
     if oslo_installed:
         def safe_func(*args, **kwargs):
-            # Lock directories use the id that is passed as either a first argument, or as a keyword.
+            # Lock directories use the id that is passed as either positional or keyword argument.
             id_parameters = [parameter_name for parameter_name in kwargs if '_id' in parameter_name]
             if len(id_parameters) == 1:
                 id_ = kwargs[id_parameters[0]]