From bef606b6ce5623c4db895a4769425d4be787ac6d Mon Sep 17 00:00:00 2001 From: Kiarash Jamali Date: Tue, 22 Aug 2023 14:37:04 +0100 Subject: [PATCH 1/3] Set default for voxel_size on inference --- model_angelo/gnn/inference.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/model_angelo/gnn/inference.py b/model_angelo/gnn/inference.py index 274ce30..c77d8bd 100644 --- a/model_angelo/gnn/inference.py +++ b/model_angelo/gnn/inference.py @@ -57,7 +57,7 @@ def infer(args): batch_converter = alphabet.get_batch_converter() lang_model = lang_model.eval() - voxel_size = args.voxel_size + voxel_size = getattr(args, "voxel_size", 1.0) rna_sequences, dna_sequences = [], [] if args.rna_fasta is not None: From baaf17a793431bc696bfca7f3ce2f82af7c486b1 Mon Sep 17 00:00:00 2001 From: Kiarash Jamali Date: Tue, 22 Aug 2023 14:37:35 +0100 Subject: [PATCH 2/3] Set default for voxel_size on inference --- model_angelo/gnn/inference_no_seq.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/model_angelo/gnn/inference_no_seq.py b/model_angelo/gnn/inference_no_seq.py index ff12d6d..ac43cf1 100644 --- a/model_angelo/gnn/inference_no_seq.py +++ b/model_angelo/gnn/inference_no_seq.py @@ -42,7 +42,7 @@ def infer(args): device_names = get_device_names(args.device) num_devices = len(device_names) - voxel_size = args.voxel_size + voxel_size = getattr(args, "voxel_size", 1.0) protein = None if args.struct.endswith("prot"): From 783418b0a04a6165bea47536df12b5cdedebbf81 Mon Sep 17 00:00:00 2001 From: Kiarash Jamali Date: Tue, 22 Aug 2023 15:07:58 +0100 Subject: [PATCH 3/3] Update __init__.py --- model_angelo/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/model_angelo/__init__.py b/model_angelo/__init__.py index 3a884dc..1a5910d 100644 --- a/model_angelo/__init__.py +++ b/model_angelo/__init__.py @@ -5,4 +5,4 @@ """ -__version__ = "1.0.1" +__version__ = "1.0.2"