risk-network 0.0.7b6__py3-none-any.whl → 0.0.7b7__py3-none-any.whl

risk/__init__.py

@@ -7,4 +7,4 @@ RISK: RISK Infers Spatial Kinships
 
 from risk.risk import RISK
 
-__version__ = "0.0.7-beta.6"
+__version__ = "0.0.7-beta.7"

risk/stats/permutation/permutation.py

@@ -4,6 +4,7 @@ risk/stats/permutation/permutation
 """
 
 from multiprocessing import get_context, Manager
+from multiprocessing.managers import ValueProxy
 from tqdm import tqdm
 from typing import Any, Callable, Dict
 
@@ -111,7 +112,6 @@ def _run_permutation_test(
     # Initialize count matrices for depletion and enrichment
     counts_depletion = np.zeros(observed_neighborhood_scores.shape)
     counts_enrichment = np.zeros(observed_neighborhood_scores.shape)
-
     # Determine the number of permutations to run in each worker process
     subset_size = num_permutations // max_workers
     remainder = num_permutations % max_workers
@@ -121,7 +121,6 @@ def _run_permutation_test(
     manager = Manager()
     progress_counter = manager.Value("i", 0)
     total_progress = num_permutations
-
     # Execute the permutation test using multiprocessing
     with ctx.Pool(max_workers) as pool:
         with tqdm(total=total_progress, desc="Total progress", position=0) as progress:
@@ -135,7 +134,8 @@ def _run_permutation_test(
                     neighborhood_score_func,
                     subset_size + (1 if i < remainder else 0),
                     progress_counter,
-                    rng,  # Pass the RNG to each process
+                    max_workers,
+                    rng,  # Pass the random number generator to each worker
                 )
                 for i in range(max_workers)
             ]
@@ -152,10 +152,10 @@ def _run_permutation_test(
             # Ensure progress bar reaches 100%
             progress.update(total_progress - progress.n)
 
-            # Accumulate results from each worker
-            for local_counts_depletion, local_counts_enrichment in results.get():
-                counts_depletion = np.add(counts_depletion, local_counts_depletion)
-                counts_enrichment = np.add(counts_enrichment, local_counts_enrichment)
+    # Accumulate results from each worker
+    for local_counts_depletion, local_counts_enrichment in results.get():
+        counts_depletion = np.add(counts_depletion, local_counts_depletion)
+        counts_enrichment = np.add(counts_enrichment, local_counts_enrichment)
 
     return counts_depletion, counts_enrichment
 
@@ -167,7 +167,8 @@ def _permutation_process_subset(
     observed_neighborhood_scores: np.ndarray,
     neighborhood_score_func: Callable,
     subset_size: int,
-    progress_counter,
+    progress_counter: ValueProxy,
+    max_workers: int,
     rng: np.random.Generator,
 ) -> tuple:
     """Process a subset of permutations for the permutation test.
@@ -179,7 +180,8 @@ def _permutation_process_subset(
         observed_neighborhood_scores (np.ndarray): Observed neighborhood scores.
         neighborhood_score_func (Callable): Function to calculate neighborhood scores.
         subset_size (int): Number of permutations to run in this subset.
-        progress_counter: Shared counter for tracking progress.
+        progress_counter (multiprocessing.managers.ValueProxy): Shared counter for tracking progress.
+        max_workers (int): Number of workers for multiprocessing.
         rng (np.random.Generator): Random number generator object.
 
     Returns:
@@ -188,10 +190,11 @@ def _permutation_process_subset(
     # Initialize local count matrices for this worker
     local_counts_depletion = np.zeros(observed_neighborhood_scores.shape)
     local_counts_enrichment = np.zeros(observed_neighborhood_scores.shape)
-    # NOTE: Limit the number of threads used by NumPy's BLAS implementation to 1.
-    # This can help prevent oversubscription of CPU resources during multiprocessing,
-    # ensuring that each process doesn't use more than one CPU core.
-    with threadpool_limits(limits=1, user_api="blas"):
+    # NOTE: Limit the number of threads used by NumPy's BLAS implementation to 1 when more than one worker is used.
+    # This can help prevent oversubscription of CPU resources during multiprocessing, ensuring that each process
+    # doesn't use more than one CPU core.
+    limits = None if max_workers == 1 else 1
+    with threadpool_limits(limits=limits, user_api="blas"):
         for _ in range(subset_size):
             # Permute the annotation matrix using the RNG
             annotation_matrix_permut = annotation_matrix[rng.permutation(idxs)]
@@ -209,7 +212,6 @@ def _permutation_process_subset(
                 local_counts_enrichment,
                 permuted_neighborhood_scores >= observed_neighborhood_scores,
             )
-
             # Update the shared progress counter
             progress_counter.value += 1
 

{risk_network-0.0.7b6.dist-info → risk_network-0.0.7b7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: risk-network
-Version: 0.0.7b6
+Version: 0.0.7b7
 Summary: A Python package for biological network analysis
 Author: Ira Horecka
 Author-email: Ira Horecka <ira89@icloud.com>

{risk_network-0.0.7b6.dist-info → risk_network-0.0.7b7.dist-info}/RECORD

@@ -1,4 +1,4 @@
-risk/__init__.py,sha256=usWMc5kXOn1-bcSacSXIi_nGKYj4cIQyRvSzvAWGbMI,112
+risk/__init__.py,sha256=0dNrjhRgqzBRG5Rqy_GKy_8ETWm10xSCYvaOm6fdI0Q,112
 risk/constants.py,sha256=XInRaH78Slnw_sWgAsBFbUHkyA0h0jL0DKGuQNbOvjM,550
 risk/risk.py,sha256=6666BzdMTgOaQl98ZKiJ19c6XBot26eTJ0iIlk-ZCZQ,20515
 risk/annotations/__init__.py,sha256=vUpVvMRE5if01Ic8QY6M2Ae3EFGJHdugEe9PdEkAW4Y,138
@@ -21,10 +21,10 @@ risk/stats/hypergeom.py,sha256=DcGYjmfcgt1qshNZPJt5IHGIHtxw9tWRS1r6QJ6V3dI,2378
 risk/stats/poisson.py,sha256=CnLk65CHViR4YhAaN3ix37iyLm_YQYGo851bSnGyyxY,1950
 risk/stats/stats.py,sha256=kvShov-94W6ffgDUTb522vB9hDJQSyTsYif_UIaFfSM,7059
 risk/stats/permutation/__init__.py,sha256=neJp7FENC-zg_CGOXqv-iIvz1r5XUKI9Ruxhmq7kDOI,105
-risk/stats/permutation/permutation.py,sha256=bFcgTJZI8cLODvGoW4QtMeBnuUs2HibJ42OZyC74Tz0,9427
+risk/stats/permutation/permutation.py,sha256=kmSZ7bQ-AD0TFiQDgIwfxTeqHa4pjp7fIcOzAqyhUNY,9714
 risk/stats/permutation/test_functions.py,sha256=HuDIM-V1jkkfE1rlaIqrWWBSKZt3dQ1f-YEDjWpnLSE,2343
-risk_network-0.0.7b6.dist-info/LICENSE,sha256=jOtLnuWt7d5Hsx6XXB2QxzrSe2sWWh3NgMfFRetluQM,35147
-risk_network-0.0.7b6.dist-info/METADATA,sha256=BQRgen5tB4jtWEWvm6VXkYPX3WudtzPpsxtFUtz3Ej0,43142
-risk_network-0.0.7b6.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
-risk_network-0.0.7b6.dist-info/top_level.txt,sha256=NX7C2PFKTvC1JhVKv14DFlFAIFnKc6Lpsu1ZfxvQwVw,5
-risk_network-0.0.7b6.dist-info/RECORD,,
+risk_network-0.0.7b7.dist-info/LICENSE,sha256=jOtLnuWt7d5Hsx6XXB2QxzrSe2sWWh3NgMfFRetluQM,35147
+risk_network-0.0.7b7.dist-info/METADATA,sha256=LDTOo8S-g09TbNgixGR4nCBkyaHW_usy0zE8kNsqTRg,43142
+risk_network-0.0.7b7.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
+risk_network-0.0.7b7.dist-info/top_level.txt,sha256=NX7C2PFKTvC1JhVKv14DFlFAIFnKc6Lpsu1ZfxvQwVw,5
+risk_network-0.0.7b7.dist-info/RECORD,,