From 206257f9f5d5c7e808e7db68d11bea5edd9e370e Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Tue, 13 Sep 2022 19:35:44 +0100 Subject: [PATCH] ai pretrain_predict: no need to plot here anymore --- aimodel/src/subcommands/pretrain_predict.py | 7 +------ 1 file changed, 1 insertion(+), 6 deletions(-) diff --git a/aimodel/src/subcommands/pretrain_predict.py b/aimodel/src/subcommands/pretrain_predict.py index bcea76a..a4e5cbf 100644 --- a/aimodel/src/subcommands/pretrain_predict.py +++ b/aimodel/src/subcommands/pretrain_predict.py @@ -74,11 +74,6 @@ def run(args): "water": water.numpy().tolist() }, separators=(',', ':'))+"\n") # Ref https://stackoverflow.com/a/64710892/1460422 - if filepath_output != "-": - handle.close() - - sys.stderr.write(">>> Plotting with UMAP\n") - filepath_output_umap = os.path.splitext(filepath_output)[0]+'.png' - vis_embeddings(filepath_output_umap, np.array([ embedding.numpy() for embedding in embeddings ])) + handle.close() sys.stderr.write(">>> Complete\n") \ No newline at end of file