Skip to content

Commit 6a22e13

Browse files
Add imports for velovi method in extvelo to support additional functionality
- Included necessary imports for torch, numpy, and scipy to enhance the velovi method. - Ensured proper integration of the velovi setup process within the extvelo function. - Maintained existing structure and comments for clarity on method testing status.
1 parent 36406ba commit 6a22e13

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

dynamo/tools/_extvelo.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -55,6 +55,7 @@ def extvelo(
5555
param_name_key=param_name_key,
5656
**kwargs
5757
)
58+
5859
return latent_data, adata
5960
elif method == "deepvelo":
6061
#tested successfully
@@ -79,6 +80,9 @@ def extvelo(
7980
elif method == "velovi":
8081
#Need to be tested
8182
from velovi import VELOVI
83+
import torch
84+
import numpy as np
85+
import scipy as sp
8286
VELOVI.setup_anndata(adata, spliced_layer=Ms_key, unspliced_layer=Mu_key)
8387
vae = VELOVI(adata)
8488
vae.train(**kwargs)

0 commit comments

Comments
 (0)