Remove torch dependency

theislab · Feb 7, 2024 · cfd4fa0 · cfd4fa0
1 parent 12e91cf
commit cfd4fa0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/scvelo/tools/_core.py b/scvelo/tools/_core.py
@@ -1,8 +1,6 @@
 from abc import abstractmethod
 from typing import NamedTuple
 
-import torch
-
 from anndata import AnnData
 
 
@@ -13,8 +11,6 @@ class _REGISTRY_KEYS_NT(NamedTuple):
 
 REGISTRY_KEYS = _REGISTRY_KEYS_NT()
 
-DEFAULT_ACTIVATION_FUNCTION = torch.nn.Softplus()
-
 
 class BaseInference:
     """Base Inference class for all velocity methods."""

diff --git a/scvelo/tools/_vi_module.py b/scvelo/tools/_vi_module.py
@@ -11,7 +11,9 @@
 from scvi.module.base import auto_move_data, BaseModuleClass, LossOutput
 from scvi.nn import Encoder, FCLayers
 
-from ._core import DEFAULT_ACTIVATION_FUNCTION, REGISTRY_KEYS
+from ._core import REGISTRY_KEYS
+
+DEFAULT_ACTIVATION_FUNCTION = torch.nn.Softplus()
 
 torch.backends.cudnn.benchmark = True