SURE-tools 2.1.88__py3-none-any.whl → 2.1.90__py3-none-any.whl

SURE/PerturbFlow.py

@@ -59,7 +59,7 @@ class PerturbFlow(nn.Module):
                  input_size: int,
                  codebook_size: int = 200,
                  cell_factor_size: int = 0,
-                 cell_factor_discrete: bool = False,
+                 cell_factor_effect_discrete: bool = False,
                  supervised_mode: bool = False,
                  z_dim: int = 10,
                  z_dist: Literal['normal','studentt','laplacian','cauchy','gumbel'] = 'gumbel',
@@ -103,7 +103,7 @@ class PerturbFlow(nn.Module):
         else:
             self.use_bias = [not zero_bias] * self.cell_factor_size
         #self.use_bias = not zero_bias
-        self.enumrate = cell_factor_discrete
+        self.enumrate = cell_factor_effect_discrete
         
         self.codebook_weights = None
 
@@ -432,8 +432,8 @@ class PerturbFlow(nn.Module):
 
             if self.cell_factor_size>0:
                 if self.enumrate:
-                    idx = torch.argmax(ns, dim=1)
-                    zn_loc = acs_loc[idx]
+                    _, ind = torch.topk(ns, 1)
+                    zn_loc = acs_loc[ind]
                     zus = self._total_effects(zn_loc, us)
                 else:
                     zus = self._total_effects(zns, us)
@@ -639,8 +639,8 @@ class PerturbFlow(nn.Module):
                 #    else:
                 #        zus = zus + self.cell_factor_effect[i]([zns,us[:,i].reshape(-1,1)])
                 if self.enumrate:
-                    idx = torch.argmax(ns, dim=1)
-                    zn_loc = acs_loc[idx]
+                    _, ind = torch.topk(ns, 1)
+                    zn_loc = acs_loc[ind]
                     zus = self._total_effects(zn_loc, us)
                 else:
                     zus = self._total_effects(zns, us)

{sure_tools-2.1.88.dist-info → sure_tools-2.1.90.dist-info}/METADATA

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

{sure_tools-2.1.88.dist-info → sure_tools-2.1.90.dist-info}/RECORD

@@ -1,4 +1,4 @@
-SURE/PerturbFlow.py,sha256=38hrb0-zx5alSVMibEpUabC5P5HCgUWYTbUz6oxvd44,55195
+SURE/PerturbFlow.py,sha256=B_v-fzSeX72rlidQnTwk169T-uwRoaPoO4INaGW_dFs,55203
 SURE/SURE.py,sha256=g8EhovBxjfpbVJA0AkmVkQ_ZW_JFc8TtkTCg8FCybV4,47750
 SURE/__init__.py,sha256=NOJI_K-eCqPgStXXvgl3wIEMp6d8saMTDYLJ7Ga9MqE,293
 SURE/assembly/__init__.py,sha256=jxZLURXKPzXe21LhrZ09LgZr33iqdjlQy4oSEj5gR2Q,172
@@ -17,9 +17,9 @@ SURE/utils/__init__.py,sha256=YF5jB-PAHJQ40OlcZ7BCZbsN2q1JKuPT6EppilRXQqM,680
 SURE/utils/custom_mlp.py,sha256=HuNb7f8-6RFjsvfEu1XOuNpLrHZkGYHgf8TpJfPSNO0,9382
 SURE/utils/queue.py,sha256=E_5PA5EWcBoGAZj8BkKQnkCK0p4C-4-xcTPqdIXaPXU,1892
 SURE/utils/utils.py,sha256=IUHjDDtYaAYllCWsZyIzqQwaLul6fJRvHRH4vIYcR-c,8462
-sure_tools-2.1.88.dist-info/licenses/LICENSE,sha256=TFHKwmrAViXQbSX5W-NDItkWFjm45HWOeUniDrqmnu0,1065
-sure_tools-2.1.88.dist-info/METADATA,sha256=0kbym3su5OEXvDQI244XPmkeGfcLRIgk6WoLe-urBeY,2678
-sure_tools-2.1.88.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-sure_tools-2.1.88.dist-info/entry_points.txt,sha256=-nJI8rVe_qqrR0HmfAODzj-JNfEqCcSsyVh6okSqyHk,83
-sure_tools-2.1.88.dist-info/top_level.txt,sha256=BtFTebdiJeqra4r6mm-uEtwVRFLZ_IjYsQ7OnalrOvY,5
-sure_tools-2.1.88.dist-info/RECORD,,
+sure_tools-2.1.90.dist-info/licenses/LICENSE,sha256=TFHKwmrAViXQbSX5W-NDItkWFjm45HWOeUniDrqmnu0,1065
+sure_tools-2.1.90.dist-info/METADATA,sha256=BkxS7WAeYZH--wcSjqNE8wL2rboHzr4NM1-9U5GnN2E,2678
+sure_tools-2.1.90.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+sure_tools-2.1.90.dist-info/entry_points.txt,sha256=-nJI8rVe_qqrR0HmfAODzj-JNfEqCcSsyVh6okSqyHk,83
+sure_tools-2.1.90.dist-info/top_level.txt,sha256=BtFTebdiJeqra4r6mm-uEtwVRFLZ_IjYsQ7OnalrOvY,5
+sure_tools-2.1.90.dist-info/RECORD,,