From 4f1775b2a2086f0091666ce77dda118fd2fd4c37 Mon Sep 17 00:00:00 2001
From: Geert-Jan Huizing <huizing@ens.fr>
Date: Tue, 7 Nov 2023 10:47:11 +0100
Subject: [PATCH] Made clear only highly_variable features are used

---
 mowgli/models.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/mowgli/models.py b/mowgli/models.py
index 9b60b78..6fb4538 100644
--- a/mowgli/models.py
+++ b/mowgli/models.py
@@ -16,6 +16,9 @@ class MowgliModel:
     Args:
         latent_dim (int, optional):
             The latent dimension of the model. Defaults to 15.
+        highly_variable (bool, optional):
+            Whether to use highly variable features. Defaults to True.
+            For now, only True is supported.
         use_mod_weight (bool, optional):
             Whether to use a different weight for each modality and each
             cell. If `True`, the weights are expected in the `mod_weight`
@@ -47,6 +50,7 @@ class MowgliModel:
     def __init__(
         self,
         latent_dim: int = 50,
+        highly_variable: bool = True,
         use_mod_weight: bool = False,
         h_regularization: float = {
             "rna": 1e-2,
@@ -65,6 +69,8 @@ def __init__(
         assert latent_dim > 0
         assert w_regularization > 0
         assert eps > 0
+        assert highly_variable is True
+        # TODO: Actually implement the use of highly_variable
 
         if isinstance(h_regularization, dict):
             for mod in h_regularization: