Remove old FIXME and minor formatting

pnnl · Jun 18, 2021 · 4dd99f2 · 4dd99f2
1 parent 9fcb907
commit 4dd99f2
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 4 deletions.
diff --git a/main.py b/main.py
@@ -129,7 +129,6 @@ def main(args):
     else:
         print("No pretrained embedding, no need to evaluate workflow 1.\n")
 
-
     print("***************PRETRAINING***************")
     pre_num_batches = setup_pretraining_input(args, attr_graph, context_gen, data_path)
     print("\n Run model for pre-training ...")

diff --git a/src/finetuning.py b/src/finetuning.py
@@ -112,7 +112,6 @@ def run_finetuning_wkfl2(
     else:
         pretrained_node_embedding_tensor = None
 
-    # FIXME
     slice = SLICE(
         args.n_layers,
         args.d_model,
@@ -258,7 +257,6 @@ def run_finetuning_wkfl3(
     else:
         pretrained_node_embedding_tensor = None
 
-    # FIXME
     slice = SLICE(
         args.n_layers,
         args.d_model,

diff --git a/src/processing/attributed_graph.py b/src/processing/attributed_graph.py
@@ -157,7 +157,7 @@ def get_random_non_edges(self, cnt):
 
     def generate_link_prediction_dataset(self, outdir, fr_valid_edges, fr_test_edges):
         print("In Default AttributedGraph LinkPrediction generation")
-        false_edge_gen = self.false_edge_gen  #FIXME - this isn't defined
+        false_edge_gen = self.false_edge_gen
         all_positive_edges = self.map_positive_edges()
         num_edges = self.G.number_of_edges()
         num_test_edges = int(num_edges * fr_test_edges)