Skip to content

Commit a972fbf

Browse files
authored
Bugfix for #178 (#179)
* Bugfix for #178 * Keep original code but check for the type of reference_sequence
1 parent c67b8b0 commit a972fbf

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

selene_sdk/predict/model_predict.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -173,7 +173,8 @@ def __init__(self,
173173
self.batch_size = batch_size
174174
self.features = features
175175
self.reference_sequence = reference_sequence
176-
if not self.reference_sequence._initialized:
176+
if type(self.reference_sequence) == Genome and \
177+
not self.reference_sequence._initialized:
177178
self.reference_sequence._unpicklable_init()
178179
if type(self.reference_sequence) == Genome and \
179180
_is_lua_trained_model(model):

0 commit comments

Comments
 (0)