lint

Author: ethanglaser

sklbench/benchmarks/sklearn_estimator.py

@@ -66,7 +66,7 @@ def get_estimator(library_name: str, estimator_name: str):
             f"Using first {classes_map[estimator_name][0]}."
         )
     estimator = classes_map[estimator_name][0]
-    #if not issubclass(estimator, BaseEstimator):
+    # if not issubclass(estimator, BaseEstimator):
     #    logger.info(f"{estimator} estimator is not derived from sklearn's BaseEstimator")
     return estimator
 
@@ -515,11 +515,14 @@ def main(bench_case: BenchCase, filters: List[BenchCase]):
     estimator_params = get_bench_case_value(
         bench_case, "algorithm:estimator_params", dict()
     )
-    #logger.debug("estimator params: " + str(estimator_params))
+    # logger.debug("estimator params: " + str(estimator_params))
     if "DBSCAN" in str(estimator_name):
         if "min_samples" in estimator_params:
             from mpi4py import MPI
-            estimator_params["min_samples"] = MPI.COMM_WORLD.Get_size() * estimator_params["min_samples"]
+
+            estimator_params["min_samples"] = (
+                MPI.COMM_WORLD.Get_size() * estimator_params["min_samples"]
+            )
     # get estimator methods for measurement
     estimator_methods = get_estimator_methods(bench_case)
 
@@ -555,7 +558,7 @@ def main(bench_case: BenchCase, filters: List[BenchCase]):
     # note: "handle" is not JSON-serializable
     if "handle" in estimator_params:
         del estimator_params["handle"]
-    #logger.debug(f"Estimator parameters:\n{custom_format(estimator_params)}")
+    # logger.debug(f"Estimator parameters:\n{custom_format(estimator_params)}")
     result_template.update(estimator_params)
 
     data_descs = {

sklbench/datasets/common.py

@@ -136,11 +136,11 @@ def cache_wrapper(**kwargs):
         data_name = kwargs["data_name"]
         data_cache = kwargs["data_cache"]
         if len(get_filenames_by_prefix(data_cache, data_name)) > 0:
-            #logger.info(f'Loading "{data_name}" dataset from cache files')
+            # logger.info(f'Loading "{data_name}" dataset from cache files')
             data = load_data_from_cache(data_cache, data_name)
             data_desc = load_data_description(data_cache, data_name)
         else:
-            #logger.info(f'Loading "{data_name}" dataset from scratch')
+            # logger.info(f'Loading "{data_name}" dataset from scratch')
             data, data_desc = function(**kwargs)
             save_data_to_cache(data, data_cache, data_name)
             save_data_description(data_desc, data_cache, data_name)

sklbench/utils/measurement.py

@@ -93,8 +93,14 @@ def measure_time(
             )
             break
     from mpi4py import MPI
+
     if MPI.COMM_WORLD.Get_rank() == 0:
-        logger.debug("iters across n runs: " + str(iters) + ", inner iters across n runs: " + str(inners))
+        logger.debug(
+            "iters across n runs: "
+            + str(iters)
+            + ", inner iters across n runs: "
+            + str(inners)
+        )
     logger.debug(times)
     # mean, std = box_filter(times)
     # if std / mean > std_mean_ratio: