Log how many processes are being used in multiprocessing

Author: jrm5100

clarite/modules/analyze/regression/glm_regression.py

@@ -88,6 +88,8 @@ def __init__(
         self.min_n = min_n
         self.report_categorical_betas = report_categorical_betas
         self.standardize_data = standardize_data
+        if self.process_num is None:
+            process_num = multiprocessing.cpu_count()
         self.process_num = process_num
 
         # Ensure the data output type is compatible
@@ -323,7 +325,8 @@ def run(self):
             else:
                 click.echo(
                     click.style(
-                        f"Running {len(rv_list):,} {rv_type} variables...", fg="green"
+                        f"Running {len(rv_list):,} {rv_type} variables using {self.process_num} processes...",
+                        fg="green",
                     )
                 )
 

clarite/modules/analyze/regression/interaction_regression.py

@@ -77,6 +77,8 @@ def __init__(
         # Custom init involving kwargs passed to this regression
         self.report_betas = report_betas
         self._process_interactions(interactions)
+        if self.process_num is None:
+            process_num = multiprocessing.cpu_count()
         self.process_num = process_num
 
         # Use a list of results instead of the default dict
@@ -212,6 +214,13 @@ def _run_interaction_regression(
 
     def run(self):
         """Run a regression object, returning the results and logging any warnings/errors"""
+        # Log how many interactions are being run using how many processes
+        click.echo(
+            click.style(
+                f"Running {len(self.interactions):,} interactions using {self.process_num} processes...",
+                fg="green",
+            )
+        )
         with multiprocessing.Pool(processes=self.process_num) as pool:
             run_result = pool.starmap(
                 self._run_interaction,

clarite/modules/analyze/regression/weighted_glm_regression.py

@@ -96,6 +96,7 @@ def __init__(
             min_n=min_n,
             report_categorical_betas=report_categorical_betas,
             standardize_data=standardize_data,
+            process_num=process_num,
         )
 
         # Custom init involving kwargs passed to this regression
@@ -283,7 +284,8 @@ def run(self):
             else:
                 click.echo(
                     click.style(
-                        f"Running {len(rv_list):,} {rv_type} variables...", fg="green"
+                        f"Running {len(rv_list):,} {rv_type} variables using {self.process_num} processes...",
+                        fg="green",
                     )
                 )