From aab870e1e36e8eceb7ef6495092454d7953e79f9 Mon Sep 17 00:00:00 2001
From: moinfar <moinfar.amirali@gmail.com>
Date: Mon, 11 Nov 2024 14:02:57 +0100
Subject: [PATCH 1/2] Improve docs
---
pyproject.toml | 1 +
src/drvi/model/__init__.py | 3 ++-
src/drvi/scvi_tools_based/model/_drvi.py | 2 +-
3 files changed, 4 insertions(+), 2 deletions(-)
diff --git a/pyproject.toml b/pyproject.toml
index fb1bd4a..fe8bf7b 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -68,6 +68,7 @@ dev = [
"twine>=4.0.2",
]
doc = [
+ "merlin-dataloader==23.8.0",
"docutils>=0.8,!=0.18.*,!=0.19.*",
"sphinx>=4",
"sphinx-book-theme>=1.0.0",
diff --git a/src/drvi/model/__init__.py b/src/drvi/model/__init__.py
index f80ec47..111a658 100644
--- a/src/drvi/model/__init__.py
+++ b/src/drvi/model/__init__.py
@@ -1,3 +1,4 @@
from drvi.scvi_tools_based.model import DRVI
+from drvi.scvi_tools_based.module import DRVIModule
-__all__ = ["DRVI"]
+__all__ = ["DRVI", "DRVIModule"]
diff --git a/src/drvi/scvi_tools_based/model/_drvi.py b/src/drvi/scvi_tools_based/model/_drvi.py
index d54a09e..1f232eb 100644
--- a/src/drvi/scvi_tools_based/model/_drvi.py
+++ b/src/drvi/scvi_tools_based/model/_drvi.py
@@ -49,7 +49,7 @@ class DRVI(VAEMixin, DRVIArchesMixin, UnsupervisedTrainingMixin, BaseModelClass,
categorical_covariates
Categorical Covariates as a list of texts. You can specify emb dimension by appending @dim to each cpvariate.
**model_kwargs
- Keyword args for :class:`~drvi.model.DRVI`
+ Keyword args for :class:`~drvi.model.DRVIModule`
Examples
--------
From a134d4ec8fc230403e7dfcab662aefa9c25032f2 Mon Sep 17 00:00:00 2001
From: moinfar <moinfar.amirali@gmail.com>
Date: Mon, 11 Nov 2024 14:09:53 +0100
Subject: [PATCH 2/2] Improve docs
---
docs/api/model.md | 1 +
1 file changed, 1 insertion(+)
diff --git a/docs/api/model.md b/docs/api/model.md
index be694c0..61d0333 100644
--- a/docs/api/model.md
+++ b/docs/api/model.md
@@ -9,4 +9,5 @@
:toctree: generated
model.DRVI
+ model.DRVIModule
```