spec bug

calico · Jun 20, 2024 · d48fb2c · d48fb2c
1 parent cf70c86
commit d48fb2c
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/src/baskerville/seqnn.py b/src/baskerville/seqnn.py
@@ -912,10 +912,11 @@ def __call__(self, x, head_i=None, dtype="float32"):
         else:
             model = self.model
 
-        if isinstance(x, np.ndarray):
-            preds = model(x).numpy().astype(dtype)
-        else:
-            preds = model(x)
+        preds = model(x).numpy().astype(dtype)
+        # if isinstance(x, np.ndarray):
+        #     preds = model(x).numpy().astype(dtype)
+        # else:
+        #     preds = model(x)
         return preds
 
     def predict(