Bugfix release 0.6.1

scarches/__init__.py

@@ -12,4 +12,4 @@
     '[email protected]'
 ])
 
-__version__ = '0.6.0'
+__version__ = '0.6.1'

scarches/models/base/_base.py

@@ -6,6 +6,7 @@
 
 import numpy as np
 import torch
+from torch.distributions import Normal
 from anndata import AnnData, read
 from scipy.sparse import issparse
 
@@ -132,12 +133,12 @@ def _load_params(cls, dir_path: str, map_location: Optional[str] = None):
         # This catches the following error:
         # RuntimeError: Attempting to deserialize object on a CUDA device
         # but torch.cuda.is_available() is False.
-        # If you are running on a CPU-only machine, please use torch.load with 
+        # If you are running on a CPU-only machine, please use torch.load with
         # map_location=torch.device('cpu') to map your storages to the CPU.
         except RuntimeError:
             with open(setup_dict_path, "rb") as handle:
                 attr_dict = UnpicklerCpu(handle).load()
-        
+
 
         model_state_dict = torch.load(model_path, map_location=map_location)
 

setup.py

@@ -9,7 +9,7 @@
     from scarches import __author__, __email__, __version__
 except ImportError:  # Deps not yet installed
     __author__ = __email__ = ''
-    __version__ = '0.6.0'
+    __version__ = '0.6.1'
 
 # otherwise readthedocs fails
 # because somewhere in the dependency tree there is the sklearn deprecated package