diff --git a/dgl_ptm/dgl_ptm/network/network_creation.py b/dgl_ptm/dgl_ptm/network/network_creation.py index fed37df..32822ff 100644 --- 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) \ No newline at end of file + return dgl.from_scipy(barabasi_albert_coo)