strand pair

calico · Jan 17, 2024 · 1c67077 · 1c67077
1 parent fc7e0c6
commit 1c67077
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/baskerville/snps.py b/src/baskerville/snps.py
@@ -68,7 +68,7 @@ def score_snps(params_file, model_file, vcf_file, worker_index, options):
 
     # load model
     if options.tensorrt:
-        seqnn_model.ensemble = OptimizedModel(model_file, params_model["strand_pair"])
+        seqnn_model.ensemble = OptimizedModel(model_file, seqnn_model.strand_pair)
         input_shape = tuple(seqnn_model.model.loaded_model_fn.inputs[0].shape.as_list())
     else:
         seqnn_model.restore(model_file)