Updated dependencies

lucas-diedrich · Jan 27, 2024 · cb8c304 · cb8c304
1 parent 3d72ec7
commit cb8c304
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -58,7 +58,7 @@ import sccoral
 # Load data
 adata = sccoral.data.splatter_simulation()
 
-# Setup model with scvi-tools syntax
+# Setup + train model with scvi-tools syntax
 sccoral.model.setup_anndata(adata,
                             categorical_covariates='categorical_covariate',
                             continuous_covariates='continuous_covariates'
@@ -67,10 +67,10 @@ model = sccoral.model.SCCORAL(adata, n_latent=7)
 model.train()
 
 # Get latent representation of cells/factor usages
-model.get_latent_representation()
+z = model.get_latent_representation()
 
 # Get interpretable gene programs (factor loadings)
-model.get_loadings()
+loadings = model.get_loadings()
 ```
 
 ## Release notes

diff --git a/pyproject.toml b/pyproject.toml
@@ -25,6 +25,8 @@ dependencies = [
     "torch>=2.0",
     # for debug logging (referenced from the issue template)
     "session-info",
+    "hyperopt",
+    "ray.tune"
 ]
 
 [project.optional-dependencies]