risk-network 0.0.12b1__tar.gz → 0.0.12b2__tar.gz

{risk_network-0.0.12b1/src/risk_network.egg-info → risk_network-0.0.12b2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: risk-network
-Version: 0.0.12b1
+Version: 0.0.12b2
 Summary: A Python package for biological network analysis
 Author-email: Ira Horecka <ira89@icloud.com>
 License: GPL-3.0-or-later

{risk_network-0.0.12b1 → risk_network-0.0.12b2}/src/risk/__init__.py

@@ -7,4 +7,4 @@ RISK: Regional Inference of Significant Kinships
 
 from risk.risk import RISK
 
-__version__ = "0.0.12-beta.1"
+__version__ = "0.0.12-beta.2"

{risk_network-0.0.12b1 → risk_network-0.0.12b2}/src/risk/network/graph/graph.py

@@ -71,14 +71,18 @@ class Graph:
         # NOTE: Only after the above attributes are initialized, we can create the summary
         self.summary = Summary(annotations, neighborhoods, self)
 
-    def pop(self, domain_id: str) -> None:
-        """Remove domain ID from instance domain ID mappings. This can be useful for cleaning up
-        domain-specific mappings based on a given criterion, as domain attributes are stored and
-        accessed only in dictionaries modified by this method.
+    def pop(self, domain_id: int) -> List[str]:
+        """Remove a domain ID from the graph and return the corresponding node labels.
 
         Args:
-            key (str): The domain ID key to be removed from each mapping.
+            key (int): The domain ID key to be removed from each mapping.
+
+        Returns:
+            List[str]: A list of node labels associated with the domain ID.
         """
+        # Get the node labels associated with the domain ID
+        node_labels = self.domain_id_to_node_labels_map.get(domain_id, [])
+
         # Define the domain mappings to be updated
         domain_mappings = [
             self.domain_id_to_node_ids_map,
@@ -97,6 +101,8 @@ class Graph:
                 domain_info["domains"].remove(domain_id)
                 domain_info["significances"].pop(domain_id)
 
+        return node_labels
+
     def _create_domain_id_to_node_ids_map(self, domains: pd.DataFrame) -> Dict[int, Any]:
         """Create a mapping from domains to the list of node IDs belonging to each domain.
 

{risk_network-0.0.12b1 → risk_network-0.0.12b2/src/risk_network.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: risk-network
-Version: 0.0.12b1
+Version: 0.0.12b2
 Summary: A Python package for biological network analysis
 Author-email: Ira Horecka <ira89@icloud.com>
 License: GPL-3.0-or-later

{risk_network-0.0.12b1 → risk_network-0.0.12b2}/src/risk_network.egg-info/top_level.txt

@@ -1 +1,2 @@
+data
 risk

{risk_network-0.0.12b1 → risk_network-0.0.12b2}/tests/test_load_graph.py

@@ -349,8 +349,12 @@ def test_pop_domain(graph):
     """
     # Define the domain ID to be removed
     domain_id_to_remove = 1
-    # Pop the domain ID
-    graph.pop(domain_id_to_remove)
+    # Retrieve expected labels before popping
+    expected_labels = graph.domain_id_to_node_labels_map.get(domain_id_to_remove)
+    # Pop the domain ID and get the returned value
+    popped_labels = graph.pop(domain_id_to_remove)
+    # Assert the returned value equals the expected labels
+    assert popped_labels == expected_labels, "Popped labels do not match the expected labels."
 
     # Check that the domain ID is removed from all relevant attributes
     assert (