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

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.


This version of SURE-tools might be problematic. Click here for more details.

SURE/PerturbFlow.py CHANGED
@@ -432,8 +432,8 @@ class PerturbFlow(nn.Module):
432
432
 
433
433
  if self.cell_factor_size>0:
434
434
  if self.enumrate:
435
- idx = torch.argmax(ns, dim=1)
436
- zn_loc = acs_loc[idx]
435
+ _, ind = torch.topk(ns, 1)
436
+ zn_loc = acs_loc[ind]
437
437
  zus = self._total_effects(zn_loc, us)
438
438
  else:
439
439
  zus = self._total_effects(zns, us)
@@ -639,8 +639,8 @@ class PerturbFlow(nn.Module):
639
639
  # else:
640
640
  # zus = zus + self.cell_factor_effect[i]([zns,us[:,i].reshape(-1,1)])
641
641
  if self.enumrate:
642
- idx = torch.argmax(ns, dim=1)
643
- zn_loc = acs_loc[idx]
642
+ _, ind = torch.topk(ns, 1)
643
+ zn_loc = acs_loc[ind]
644
644
  zus = self._total_effects(zn_loc, us)
645
645
  else:
646
646
  zus = self._total_effects(zns, us)
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: SURE-tools
3
- Version: 2.1.89
3
+ Version: 2.1.90
4
4
  Summary: Succinct Representation of Single Cells
5
5
  Home-page: https://github.com/ZengFLab/SURE
6
6
  Author: Feng Zeng
@@ -1,4 +1,4 @@
1
- SURE/PerturbFlow.py,sha256=9rVXYkA4A2OhX7hgyfhNuX-oQzlQWCuSmc-76UHL0NM,55209
1
+ SURE/PerturbFlow.py,sha256=B_v-fzSeX72rlidQnTwk169T-uwRoaPoO4INaGW_dFs,55203
2
2
  SURE/SURE.py,sha256=g8EhovBxjfpbVJA0AkmVkQ_ZW_JFc8TtkTCg8FCybV4,47750
3
3
  SURE/__init__.py,sha256=NOJI_K-eCqPgStXXvgl3wIEMp6d8saMTDYLJ7Ga9MqE,293
4
4
  SURE/assembly/__init__.py,sha256=jxZLURXKPzXe21LhrZ09LgZr33iqdjlQy4oSEj5gR2Q,172
@@ -17,9 +17,9 @@ SURE/utils/__init__.py,sha256=YF5jB-PAHJQ40OlcZ7BCZbsN2q1JKuPT6EppilRXQqM,680
17
17
  SURE/utils/custom_mlp.py,sha256=HuNb7f8-6RFjsvfEu1XOuNpLrHZkGYHgf8TpJfPSNO0,9382
18
18
  SURE/utils/queue.py,sha256=E_5PA5EWcBoGAZj8BkKQnkCK0p4C-4-xcTPqdIXaPXU,1892
19
19
  SURE/utils/utils.py,sha256=IUHjDDtYaAYllCWsZyIzqQwaLul6fJRvHRH4vIYcR-c,8462
20
- sure_tools-2.1.89.dist-info/licenses/LICENSE,sha256=TFHKwmrAViXQbSX5W-NDItkWFjm45HWOeUniDrqmnu0,1065
21
- sure_tools-2.1.89.dist-info/METADATA,sha256=PwP97S_sckAYLFgasDYJ9zbCV_FRNSa1LEFSl46XF-g,2678
22
- sure_tools-2.1.89.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
23
- sure_tools-2.1.89.dist-info/entry_points.txt,sha256=-nJI8rVe_qqrR0HmfAODzj-JNfEqCcSsyVh6okSqyHk,83
24
- sure_tools-2.1.89.dist-info/top_level.txt,sha256=BtFTebdiJeqra4r6mm-uEtwVRFLZ_IjYsQ7OnalrOvY,5
25
- sure_tools-2.1.89.dist-info/RECORD,,
20
+ sure_tools-2.1.90.dist-info/licenses/LICENSE,sha256=TFHKwmrAViXQbSX5W-NDItkWFjm45HWOeUniDrqmnu0,1065
21
+ sure_tools-2.1.90.dist-info/METADATA,sha256=BkxS7WAeYZH--wcSjqNE8wL2rboHzr4NM1-9U5GnN2E,2678
22
+ sure_tools-2.1.90.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
23
+ sure_tools-2.1.90.dist-info/entry_points.txt,sha256=-nJI8rVe_qqrR0HmfAODzj-JNfEqCcSsyVh6okSqyHk,83
24
+ sure_tools-2.1.90.dist-info/top_level.txt,sha256=BtFTebdiJeqra4r6mm-uEtwVRFLZ_IjYsQ7OnalrOvY,5
25
+ sure_tools-2.1.90.dist-info/RECORD,,