Skip to content
Snippets Groups Projects
Commit c8f51e83 authored by dimitrova's avatar dimitrova
Browse files

small fix

parent d091f7dd
No related branches found
No related tags found
No related merge requests found
......@@ -32,6 +32,7 @@ def get_num_random_nodes_for_all_classes_read(path = "/home/utaemon/SP/graph.pkl
#unused
'''
def get_num_of_random_nodes(path = "/home/utaemon/SP/graph.pkl", seed=0, num = 3):
"""Get random nodes."""
random.seed(seed)
......@@ -40,4 +41,5 @@ def get_num_of_random_nodes(path = "/home/utaemon/SP/graph.pkl", seed=0, num = 3
def pickle_output(method = get_num_random_nodes_for_all_classes_read(), output = "random_nodes.pkl"):
with open(output, "wb") as file:
pkl.dump(method, file)
\ No newline at end of file
pkl.dump(method, file)
'''
\ No newline at end of file
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment