SURE-tools 2.1.88__tar.gz → 2.1.90__tar.gz

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.

Files changed (30) hide show
  1. {sure_tools-2.1.88 → sure_tools-2.1.90}/PKG-INFO +1 -1
  2. {sure_tools-2.1.88 → sure_tools-2.1.90}/SURE/PerturbFlow.py +6 -6
  3. {sure_tools-2.1.88 → sure_tools-2.1.90}/SURE_tools.egg-info/PKG-INFO +1 -1
  4. {sure_tools-2.1.88 → sure_tools-2.1.90}/setup.py +1 -1
  5. {sure_tools-2.1.88 → sure_tools-2.1.90}/LICENSE +0 -0
  6. {sure_tools-2.1.88 → sure_tools-2.1.90}/README.md +0 -0
  7. {sure_tools-2.1.88 → sure_tools-2.1.90}/SURE/SURE.py +0 -0
  8. {sure_tools-2.1.88 → sure_tools-2.1.90}/SURE/__init__.py +0 -0
  9. {sure_tools-2.1.88 → sure_tools-2.1.90}/SURE/assembly/__init__.py +0 -0
  10. {sure_tools-2.1.88 → sure_tools-2.1.90}/SURE/assembly/assembly.py +0 -0
  11. {sure_tools-2.1.88 → sure_tools-2.1.90}/SURE/assembly/atlas.py +0 -0
  12. {sure_tools-2.1.88 → sure_tools-2.1.90}/SURE/atac/__init__.py +0 -0
  13. {sure_tools-2.1.88 → sure_tools-2.1.90}/SURE/atac/utils.py +0 -0
  14. {sure_tools-2.1.88 → sure_tools-2.1.90}/SURE/codebook/__init__.py +0 -0
  15. {sure_tools-2.1.88 → sure_tools-2.1.90}/SURE/codebook/codebook.py +0 -0
  16. {sure_tools-2.1.88 → sure_tools-2.1.90}/SURE/flow/__init__.py +0 -0
  17. {sure_tools-2.1.88 → sure_tools-2.1.90}/SURE/flow/flow_stats.py +0 -0
  18. {sure_tools-2.1.88 → sure_tools-2.1.90}/SURE/flow/plot_quiver.py +0 -0
  19. {sure_tools-2.1.88 → sure_tools-2.1.90}/SURE/perturb/__init__.py +0 -0
  20. {sure_tools-2.1.88 → sure_tools-2.1.90}/SURE/perturb/perturb.py +0 -0
  21. {sure_tools-2.1.88 → sure_tools-2.1.90}/SURE/utils/__init__.py +0 -0
  22. {sure_tools-2.1.88 → sure_tools-2.1.90}/SURE/utils/custom_mlp.py +0 -0
  23. {sure_tools-2.1.88 → sure_tools-2.1.90}/SURE/utils/queue.py +0 -0
  24. {sure_tools-2.1.88 → sure_tools-2.1.90}/SURE/utils/utils.py +0 -0
  25. {sure_tools-2.1.88 → sure_tools-2.1.90}/SURE_tools.egg-info/SOURCES.txt +0 -0
  26. {sure_tools-2.1.88 → sure_tools-2.1.90}/SURE_tools.egg-info/dependency_links.txt +0 -0
  27. {sure_tools-2.1.88 → sure_tools-2.1.90}/SURE_tools.egg-info/entry_points.txt +0 -0
  28. {sure_tools-2.1.88 → sure_tools-2.1.90}/SURE_tools.egg-info/requires.txt +0 -0
  29. {sure_tools-2.1.88 → sure_tools-2.1.90}/SURE_tools.egg-info/top_level.txt +0 -0
  30. {sure_tools-2.1.88 → sure_tools-2.1.90}/setup.cfg +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: SURE-tools
3
- Version: 2.1.88
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
@@ -59,7 +59,7 @@ class PerturbFlow(nn.Module):
59
59
  input_size: int,
60
60
  codebook_size: int = 200,
61
61
  cell_factor_size: int = 0,
62
- cell_factor_discrete: bool = False,
62
+ cell_factor_effect_discrete: bool = False,
63
63
  supervised_mode: bool = False,
64
64
  z_dim: int = 10,
65
65
  z_dist: Literal['normal','studentt','laplacian','cauchy','gumbel'] = 'gumbel',
@@ -103,7 +103,7 @@ class PerturbFlow(nn.Module):
103
103
  else:
104
104
  self.use_bias = [not zero_bias] * self.cell_factor_size
105
105
  #self.use_bias = not zero_bias
106
- self.enumrate = cell_factor_discrete
106
+ self.enumrate = cell_factor_effect_discrete
107
107
 
108
108
  self.codebook_weights = None
109
109
 
@@ -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.88
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
@@ -5,7 +5,7 @@ with open("README.md", "r") as fh:
5
5
 
6
6
  setup(
7
7
  name='SURE-tools',
8
- version='2.1.88',
8
+ version='2.1.90',
9
9
  description='Succinct Representation of Single Cells',
10
10
  long_description=long_description,
11
11
  long_description_content_type="text/markdown",
File without changes
File without changes
File without changes
File without changes