m

EdwardCuiPeacock · Apr 13, 2021 · 9439d9b · 9439d9b
1 parent 2580673
commit 9439d9b
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/models/node2vec/node2vec.py b/models/node2vec/node2vec.py
@@ -3,7 +3,6 @@
 from absl import logging
 import os
 import gc
-import time
 from typing import Text, Optional, Union, Callable, List
 import numpy as np
 from tqdm import tqdm
@@ -486,7 +485,7 @@ def parse_tensor_f(x):
         ds = tf.data.Dataset.from_tensor_slices(features).map(tf.io.serialize_tensor)
         writer = tf.data.experimental.TFRecordWriter(data_uri)
         writer.write(ds)
-        
+
     return data_uri_list, num_rows_saved