SURE-tools 2.1.2__py3-none-any.whl → 2.1.4__py3-none-any.whl

SURE/PerturbFlow.py

@@ -702,11 +702,11 @@ class PerturbFlow(nn.Module):
         cb = tensor_to_numpy(cb)
         return cb
 
-    def _get_cell_embedding(self, xs):           
+    def _get_basis_embedding(self, xs):           
         zns, _ = self.encoder_zn(xs)
         return zns 
     
-    def get_cell_embedding(self, 
+    def get_basis_embedding(self, 
                              xs, 
                              batch_size: int = 1024):
         """
@@ -868,7 +868,7 @@ class PerturbFlow(nn.Module):
         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 = self._get_expression_response(delta_Z_batch)
                 R.append(tensor_to_numpy(r))
                 pbar.update(1)
 

{sure_tools-2.1.2.dist-info → sure_tools-2.1.4.dist-info}/METADATA

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

{sure_tools-2.1.2.dist-info → sure_tools-2.1.4.dist-info}/RECORD

@@ -1,4 +1,4 @@
-SURE/PerturbFlow.py,sha256=RQoIhYQJdQpHdY_sMeDuqurbwvm6IX1XH7SVWG6SmS0,51658
+SURE/PerturbFlow.py,sha256=M_h66aHWTl5Vf0TC6e-UJrC4T3FQGH0wFLZ95jgdt7U,51674
 SURE/SURE.py,sha256=_ZOymj24DLQju0Lb90lKspHPmqIUDDzjIEr9t4qgqCI,48364
 SURE/SURE2.py,sha256=8wlnMwb1xuf9QUksNkWdWx5ZWq-xIy9NLx8RdUnE82o,48501
 SURE/__init__.py,sha256=xV10iBbh69g4mjBMb1cQxjuHe8e3Aq7pDzkZmx5G754,260
@@ -17,9 +17,9 @@ SURE/utils/__init__.py,sha256=Htqv4KqVKcRiaaTBsR-6yZ4LSlbhbzutjNKXGD9-uds,660
 SURE/utils/custom_mlp.py,sha256=07TYX1HgxfEjb_3i5MpiZfNhOhx3dKntuwGkrpteWiM,7036
 SURE/utils/queue.py,sha256=E_5PA5EWcBoGAZj8BkKQnkCK0p4C-4-xcTPqdIXaPXU,1892
 SURE/utils/utils.py,sha256=IUHjDDtYaAYllCWsZyIzqQwaLul6fJRvHRH4vIYcR-c,8462
-sure_tools-2.1.2.dist-info/licenses/LICENSE,sha256=TFHKwmrAViXQbSX5W-NDItkWFjm45HWOeUniDrqmnu0,1065
-sure_tools-2.1.2.dist-info/METADATA,sha256=30k-_5ftI_F5fd76yo792KMd2rjISQy-8PFi-1YuOrI,2650
-sure_tools-2.1.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-sure_tools-2.1.2.dist-info/entry_points.txt,sha256=-nJI8rVe_qqrR0HmfAODzj-JNfEqCcSsyVh6okSqyHk,83
-sure_tools-2.1.2.dist-info/top_level.txt,sha256=BtFTebdiJeqra4r6mm-uEtwVRFLZ_IjYsQ7OnalrOvY,5
-sure_tools-2.1.2.dist-info/RECORD,,
+sure_tools-2.1.4.dist-info/licenses/LICENSE,sha256=TFHKwmrAViXQbSX5W-NDItkWFjm45HWOeUniDrqmnu0,1065
+sure_tools-2.1.4.dist-info/METADATA,sha256=oJNTuJ2eturhVARGLQkmZ3-wGhGmWTTMAOXHABKUhJg,2650
+sure_tools-2.1.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+sure_tools-2.1.4.dist-info/entry_points.txt,sha256=-nJI8rVe_qqrR0HmfAODzj-JNfEqCcSsyVh6okSqyHk,83
+sure_tools-2.1.4.dist-info/top_level.txt,sha256=BtFTebdiJeqra4r6mm-uEtwVRFLZ_IjYsQ7OnalrOvY,5
+sure_tools-2.1.4.dist-info/RECORD,,