SURE-tools 3.7.1__tar.gz → 3.7.2__tar.gz

{sure_tools-3.7.1 → sure_tools-3.7.2}/PKG-INFO

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

{sure_tools-3.7.1 → sure_tools-3.7.2}/SURE/SURE_vae.py

@@ -269,7 +269,6 @@ class SUREVAE(nn.Module):
                     )
                 )
             
-            self.encoder_condition_effects = self.decoder_condition_effects
             '''self.encoder_condition_effects = nn.ModuleList()
             for condition_size in self.condition_sizes:
                 self.encoder_condition_effects.append(ZeroBiasMLP3(
@@ -479,7 +478,7 @@ class SUREVAE(nn.Module):
             shift = 0
             for i, condition_size in enumerate(self.condition_sizes):
                 cs_i = cs[:,shift:(shift+condition_size)]
-                l,s = self.encoder_condition_effects[i]([cs_i,zns])
+                l,s = self.decoder_condition_effects[i]([cs_i,zns])
                 zcs_i = pyro.sample(f'zcs_{i}', dist.Normal(l, s).to_event(1))
                 shift += condition_size
                     
@@ -627,7 +626,7 @@ class SUREVAE(nn.Module):
                 C_batch = cs[idx].to(self.get_device())
                 
                 z_basal = self._get_cell_embedding(X_batch)
-                dzs,_ = self.encoder_condition_effects[i]([C_batch,z_basal])
+                dzs,_ = self.decoder_condition_effects[i]([C_batch,z_basal])
                 
                 A.append(tensor_to_numpy(dzs))
                 pbar.update(1)
@@ -671,7 +670,7 @@ class SUREVAE(nn.Module):
                 shift = 0
                 for i, condition_size in enumerate(self.condition_sizes):
                     C_batch_i = C_batch[:, shift:(shift+condition_size)]
-                    zcs_i,_ = self.encoder_condition_effects[i]([C_batch_i,z_basal])
+                    zcs_i,_ = self.decoder_condition_effects[i]([C_batch_i,z_basal])
                     zcs += zcs_i
                     shift += condition_size
                     

{sure_tools-3.7.1 → sure_tools-3.7.2}/SURE_tools.egg-info/PKG-INFO

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

{sure_tools-3.7.1 → sure_tools-3.7.2}/setup.py

@@ -5,7 +5,7 @@ with open("README.md", "r") as fh:
 
 setup(
     name='SURE-tools',
-    version='3.7.1',
+    version='3.7.2',
     description='Succinct Representation of Single Cells',
     long_description=long_description,
     long_description_content_type="text/markdown",