Update network_creation.py

SDCCA · Aug 20, 2024 · 59d8294 · 59d8294
1 parent 3c8eae4
commit 59d8294
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/dgl_ptm/dgl_ptm/network/network_creation.py b/dgl_ptm/dgl_ptm/network/network_creation.py
@@ -4,6 +4,7 @@
 import dgl
 import networkx as nx
 import random
+import torch
 
 # network_creation - Creates the network between the initialized nodes using edges from DGL
 
@@ -25,7 +26,7 @@ def network_creation(num_agents, method, **kwargs):
             seed  = kwargs['seed']
             print(f"using seed {seed} for network creation.")
         else:
-            seed = random.randint(1, 100000)
+            seed = torch.initial_seed() 
 
         if 'new_node_edges' in kwargs.keys(): 
             new_node_edges = kwargs['new_node_edges']
@@ -58,4 +59,4 @@ def barabasi_albert_graph(num_agents, new_node_edges=1, seed=1):
     print(new_node_edges, seed)
 
     #Return DGL graph from networkx graph
-    return dgl.from_scipy(barabasi_albert_coo)
+    return dgl.from_scipy(barabasi_albert_coo)