Skip to content
This repository was archived by the owner on Jan 18, 2023. It is now read-only.

Commit 7a9b0cb

Browse files
adamgayosoadamgayoso
committed
fix import
1 parent f9d8746 commit 7a9b0cb

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

‎mypackage/_mypyromodel.py‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,8 +5,8 @@
55
import torch
66
from anndata import AnnData
77
from scvi.dataloaders import DataSplitter
8-
from scvi.lightning import PyroTrainingPlan
9-
from scvi.model.base import BaseModelClass, TrainRunner
8+
from scvi.model.base import BaseModelClass
9+
from scvi.train import PyroTrainingPlan, TrainRunner
1010

1111
from ._mypyromodule import MyPyroModule
1212

0 commit comments

Comments
 (0)