Update scvi/external/sysvi/_base_components.py

Hrovatin · Mar 15, 2024 · c5f5c37 · c5f5c37
1 parent 8c25dba
commit c5f5c37
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scvi/external/sysvi/_base_components.py b/scvi/external/sysvi/_base_components.py
@@ -30,7 +30,7 @@ class Embedding(Module):
         Apply layer normalization
     """
 
-    def __init__(self, size, cov_embed_dims: int = 10, normalize: bool = True):
+    def __init__(self, size: int, cov_embed_dims: int = 10, normalize: bool = True):
         super().__init__()
 
         self.normalize = normalize