fix(models): mark apparently unused sample sites

Signed-off-by: Cameron Smith <[email protected]>
pinellolab · Sep 13, 2024 · 4684551 · 4684551
1 parent 1bb5b24
commit 4684551
Showing 1 changed file with 13 additions and 6 deletions.
diff --git a/src/pyrovelocity/models/_velocity_model.py b/src/pyrovelocity/models/_velocity_model.py
@@ -167,10 +167,12 @@ def u_scale(self):
     def s_scale(self):
         return self._pyrosample_helper(0.1)
 
+    # TODO: remove unused
     @PyroSample
     def u_inf(self):
         return self._pyrosample_helper(0.1)
 
+    # TODO: remove unused
     @PyroSample
     def s_inf(self):
         return self._pyrosample_helper(0.1)
@@ -179,14 +181,17 @@ def s_inf(self):
     def dt_switching(self):
         return self._pyrosample_helper(1.0)
 
+    # TODO: remove unused
     @PyroSample
     def gene_offset(self):
         return Normal(self.zero, self.one)
 
+    # TODO: remove unused
     @PyroSample
     def t_scale(self):
         return Normal(self.zero, self.one * 0.1)
 
+    # TODO: remove unused
     @PyroSample
     def latent_time(self):
         if self.shared_time & self.plate_size == 2:
@@ -198,6 +203,7 @@ def latent_time(self):
                 .to_event(1)
             )
 
+    # TODO: remove unused
     @PyroSample
     def cell_time(self):
         if self.plate_size == 2 and self.shared_time:
@@ -208,12 +214,13 @@ def cell_time(self):
     def _pyrosample_helper(self, scale: float):
         if self.plate_size == 2:
             return LogNormal(self.zero, self.one * scale)
-        return (
-            LogNormal(self.zero, self.one * 0.1)
-            .expand((self.num_genes,))
-            .to_event(1)
-            .mask(False)
-        )
+        else:
+            return (
+                LogNormal(self.zero, self.one * 0.1)
+                .expand((self.num_genes,))
+                .to_event(1)
+                .mask(False)
+            )
 
     @beartype
     def get_likelihood(