SURE-tools 2.0.9__tar.gz → 2.0.10__tar.gz

{sure_tools-2.0.9 → sure_tools-2.0.10}/PKG-INFO

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

{sure_tools-2.0.9 → sure_tools-2.0.10}/SURE/SURE.py

@@ -860,6 +860,30 @@ class SURE(nn.Module):
         ms = self.cell_factor_effect[factor_idx]([zs,ps])
         return tensor_to_numpy(ms)
     
+    def _get_expression_responses(self, delta_zs):
+        return self.decoder_concentrate(delta_zs)
+    
+    def get_expression_responses(self, 
+                             delta_zs, 
+                             batch_size: int = 1024):
+        """
+        Return cells' changes in the latent space induced by specific perturbation of a factor
+
+        """
+        delta_zs = convert_to_tensor(delta_zs, device=self.get_device())
+        dataset = CustomDataset(delta_zs)
+        dataloader = DataLoader(dataset, batch_size=batch_size, shuffle=False)
+
+        R = []
+        with tqdm(total=len(dataloader), desc='', unit='batch') as pbar:
+            for delta_Z_batch, _ in dataloader:
+                r = self._cell_move(delta_Z_batch)
+                R.append(tensor_to_numpy(r))
+                pbar.update(1)
+
+        R = np.concatenate(R)
+        return R
+    
     def preprocess(self, xs, threshold=0):
         if self.loss_func == 'bernoulli':
             ad = sc.AnnData(xs)

{sure_tools-2.0.9 → sure_tools-2.0.10}/SURE_tools.egg-info/PKG-INFO

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

{sure_tools-2.0.9 → sure_tools-2.0.10}/setup.py

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