Minor improvements

cjayb · May 19, 2021 · 43e97d6 · 43e97d6
1 parent f2ed3de
commit 43e97d6
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/hnn_core/network.py b/hnn_core/network.py
@@ -1100,11 +1100,13 @@ def add_connection(self, src_gids, target_gids, loc, receptor,
 
         # Probabilistically define connections
         if probability != 1.0:
-            self.connection_probability(conn, probability)
+            self._connection_probability(conn, probability)
+
+        conn['probability'] = probability
 
         self.connectivity.append(deepcopy(conn))
 
-    def connection_probability(self, conn, probability):
+    def _connection_probability(self, conn, probability):
         """Remove/keep a random subset of connections.
 
         Parameters
@@ -1154,8 +1156,6 @@ def connection_probability(self, conn, probability):
         for src_gid in remove_srcs:
             conn['gid_pairs'].pop(src_gid)
 
-        conn['probability'] = probability
-
     def clear_connectivity(self):
         """Remove all connections defined in Network.connectivity_list"""
         self.connectivity = list()