SURE-tools 4.0.4__py3-none-any.whl → 4.0.6__py3-none-any.whl

SURE/SURE_nsf.py

@@ -424,7 +424,7 @@ class SURENF(nn.Module):
                 zps = self.perturb_effect([ps, zs+zcs])
             else:
                 zps = torch.zeros_like(zs)
-            if (self.covariate_sizes>0) and (fs is not None):
+            if (np.sum(self.covariate_sizes)>0) and (fs is not None):
                 zfs = torch.zeros_like(zs)
                 shift = 0
                 for i, covariate_size in enumerate(self.covariate_sizes):

SURE/SURE_vae.py

@@ -444,7 +444,7 @@ class SUREVAE(nn.Module):
             else:
                 zfs = torch.zeros_like(zs)
 
-            log_mu = self.decoder_log_mu(zs+zcs,zps,zfs)
+            log_mu = self.decoder_log_mu([zs+zcs,zps,zfs])
             if self.loss_func in ['bernoulli']:
                 log_theta = log_mu
             elif self.loss_func in ['negbinomial']:
@@ -707,7 +707,7 @@ class SUREVAE(nn.Module):
                     
                 zfs = torch.zeros_like(z_basal)
                 
-                log_mu = self.decoder_log_mu(z_basal+zcs,zps,zfs)
+                log_mu = self.decoder_log_mu([z_basal+zcs,zps,zfs])
                 if self.loss_func == 'bernoulli':
                     counts = dist.Bernoulli(logits=log_mu).to_event(1).mean
                 else:

{sure_tools-4.0.4.dist-info → sure_tools-4.0.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: SURE-tools
-Version: 4.0.4
+Version: 4.0.6
 Summary: Succinct Representation of Single Cells
 Home-page: https://github.com/ZengFLab/SURE
 Author: Feng Zeng

{sure_tools-4.0.4.dist-info → sure_tools-4.0.6.dist-info}/RECORD

@@ -1,7 +1,7 @@
 SURE/SURE.py,sha256=YcJ6Lxt4ziQXZ5Yk02Ss-fJRl_M1Z0QghLY_yeHt5Pc,16659
 SURE/SUREMO.py,sha256=hN0G0ZEBNQdmj0gGlBIy1wjKGKqAMUemDjQeICzvNUY,47644
-SURE/SURE_nsf.py,sha256=uKpau3nDdlX58lLzPzecZULKobyOYB7xKtcYo6vwbak,49541
-SURE/SURE_vae.py,sha256=GY7r5Yy2U8GlI_QvBb43z4r0Ez0NGYqWebWICLObPNs,46407
+SURE/SURE_nsf.py,sha256=otCsvCEYdKZbzOZ1SLOtjA89nFE0Dw8WRQSH_86LaWk,49549
+SURE/SURE_vae.py,sha256=VhJk9K_RA6IzaZ8WpP8T3j1c5I0_o7x6VZzyP_QBseI,46411
 SURE/SURE_vanilla.py,sha256=I1RHHCpzk8ml1vMdH_gITOzAFXrHYUA7IAwbVSoxmBo,27327
 SURE/__init__.py,sha256=eJN0vlGblWir1JHaoiQqbQHzS_C1PNYcA_ls7UviTqc,444
 SURE/atac/__init__.py,sha256=3smP8IKHfwNCd1G_sZH3pKHXuLkLpFuLtjUTUSy7_As,34
@@ -15,8 +15,8 @@ SURE/utils/custom_mlp.py,sha256=XvviNUYeaZ5D1evqETyWLxgtL56mmFQR_pxsYpKw0yY,1170
 SURE/utils/label.py,sha256=joKO1mSkjZXeLvSeC7GluQk4-_qgGgPqlwWixdcbKMQ,4648
 SURE/utils/queue.py,sha256=E_5PA5EWcBoGAZj8BkKQnkCK0p4C-4-xcTPqdIXaPXU,1892
 SURE/utils/utils.py,sha256=IUHjDDtYaAYllCWsZyIzqQwaLul6fJRvHRH4vIYcR-c,8462
-sure_tools-4.0.4.dist-info/licenses/LICENSE,sha256=TFHKwmrAViXQbSX5W-NDItkWFjm45HWOeUniDrqmnu0,1065
-sure_tools-4.0.4.dist-info/METADATA,sha256=hzyrZERvWqbEnMRTap91Hk7niBAiJ1zBej0uJQeUdnY,1661
-sure_tools-4.0.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-sure_tools-4.0.4.dist-info/top_level.txt,sha256=BtFTebdiJeqra4r6mm-uEtwVRFLZ_IjYsQ7OnalrOvY,5
-sure_tools-4.0.4.dist-info/RECORD,,
+sure_tools-4.0.6.dist-info/licenses/LICENSE,sha256=TFHKwmrAViXQbSX5W-NDItkWFjm45HWOeUniDrqmnu0,1065
+sure_tools-4.0.6.dist-info/METADATA,sha256=tOIma_c_X8sYanUli5qEv4qWe8WG2ohoKBjrrhlpeIc,1661
+sure_tools-4.0.6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+sure_tools-4.0.6.dist-info/top_level.txt,sha256=BtFTebdiJeqra4r6mm-uEtwVRFLZ_IjYsQ7OnalrOvY,5
+sure_tools-4.0.6.dist-info/RECORD,,