SURE-tools 2.1.90__tar.gz → 2.1.92__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.
Files changed (30) hide show
  1. {sure_tools-2.1.90 → sure_tools-2.1.92}/PKG-INFO +1 -1
  2. {sure_tools-2.1.90 → sure_tools-2.1.92}/SURE/PerturbFlow.py +2 -14
  3. {sure_tools-2.1.90 → sure_tools-2.1.92}/SURE_tools.egg-info/PKG-INFO +1 -1
  4. {sure_tools-2.1.90 → sure_tools-2.1.92}/setup.py +1 -1
  5. {sure_tools-2.1.90 → sure_tools-2.1.92}/LICENSE +0 -0
  6. {sure_tools-2.1.90 → sure_tools-2.1.92}/README.md +0 -0
  7. {sure_tools-2.1.90 → sure_tools-2.1.92}/SURE/SURE.py +0 -0
  8. {sure_tools-2.1.90 → sure_tools-2.1.92}/SURE/__init__.py +0 -0
  9. {sure_tools-2.1.90 → sure_tools-2.1.92}/SURE/assembly/__init__.py +0 -0
  10. {sure_tools-2.1.90 → sure_tools-2.1.92}/SURE/assembly/assembly.py +0 -0
  11. {sure_tools-2.1.90 → sure_tools-2.1.92}/SURE/assembly/atlas.py +0 -0
  12. {sure_tools-2.1.90 → sure_tools-2.1.92}/SURE/atac/__init__.py +0 -0
  13. {sure_tools-2.1.90 → sure_tools-2.1.92}/SURE/atac/utils.py +0 -0
  14. {sure_tools-2.1.90 → sure_tools-2.1.92}/SURE/codebook/__init__.py +0 -0
  15. {sure_tools-2.1.90 → sure_tools-2.1.92}/SURE/codebook/codebook.py +0 -0
  16. {sure_tools-2.1.90 → sure_tools-2.1.92}/SURE/flow/__init__.py +0 -0
  17. {sure_tools-2.1.90 → sure_tools-2.1.92}/SURE/flow/flow_stats.py +0 -0
  18. {sure_tools-2.1.90 → sure_tools-2.1.92}/SURE/flow/plot_quiver.py +0 -0
  19. {sure_tools-2.1.90 → sure_tools-2.1.92}/SURE/perturb/__init__.py +0 -0
  20. {sure_tools-2.1.90 → sure_tools-2.1.92}/SURE/perturb/perturb.py +0 -0
  21. {sure_tools-2.1.90 → sure_tools-2.1.92}/SURE/utils/__init__.py +0 -0
  22. {sure_tools-2.1.90 → sure_tools-2.1.92}/SURE/utils/custom_mlp.py +0 -0
  23. {sure_tools-2.1.90 → sure_tools-2.1.92}/SURE/utils/queue.py +0 -0
  24. {sure_tools-2.1.90 → sure_tools-2.1.92}/SURE/utils/utils.py +0 -0
  25. {sure_tools-2.1.90 → sure_tools-2.1.92}/SURE_tools.egg-info/SOURCES.txt +0 -0
  26. {sure_tools-2.1.90 → sure_tools-2.1.92}/SURE_tools.egg-info/dependency_links.txt +0 -0
  27. {sure_tools-2.1.90 → sure_tools-2.1.92}/SURE_tools.egg-info/entry_points.txt +0 -0
  28. {sure_tools-2.1.90 → sure_tools-2.1.92}/SURE_tools.egg-info/requires.txt +0 -0
  29. {sure_tools-2.1.90 → sure_tools-2.1.92}/SURE_tools.egg-info/top_level.txt +0 -0
  30. {sure_tools-2.1.90 → sure_tools-2.1.92}/setup.cfg +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: SURE-tools
3
- Version: 2.1.90
3
+ Version: 2.1.92
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,6 @@ 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_effect_discrete: bool = False,
63
62
  supervised_mode: bool = False,
64
63
  z_dim: int = 10,
65
64
  z_dist: Literal['normal','studentt','laplacian','cauchy','gumbel'] = 'gumbel',
@@ -103,7 +102,6 @@ class PerturbFlow(nn.Module):
103
102
  else:
104
103
  self.use_bias = [not zero_bias] * self.cell_factor_size
105
104
  #self.use_bias = not zero_bias
106
- self.enumrate = cell_factor_effect_discrete
107
105
 
108
106
  self.codebook_weights = None
109
107
 
@@ -431,12 +429,7 @@ class PerturbFlow(nn.Module):
431
429
  zns = pyro.sample('zn', dist.Gumbel(zn_loc, zn_scale).to_event(1))
432
430
 
433
431
  if self.cell_factor_size>0:
434
- if self.enumrate:
435
- _, ind = torch.topk(ns, 1)
436
- zn_loc = acs_loc[ind]
437
- zus = self._total_effects(zn_loc, us)
438
- else:
439
- zus = self._total_effects(zns, us)
432
+ zus = self._total_effects(zns, us)
440
433
  zs = zns+zus
441
434
  else:
442
435
  zs = zns
@@ -638,12 +631,7 @@ class PerturbFlow(nn.Module):
638
631
  # zus = self.cell_factor_effect[i]([zns,us[:,i].reshape(-1,1)])
639
632
  # else:
640
633
  # zus = zus + self.cell_factor_effect[i]([zns,us[:,i].reshape(-1,1)])
641
- if self.enumrate:
642
- _, ind = torch.topk(ns, 1)
643
- zn_loc = acs_loc[ind]
644
- zus = self._total_effects(zn_loc, us)
645
- else:
646
- zus = self._total_effects(zns, us)
634
+ zus = self._total_effects(zns, us)
647
635
  zs = zns+zus
648
636
  else:
649
637
  zs = zns
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: SURE-tools
3
- Version: 2.1.90
3
+ Version: 2.1.92
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.90',
8
+ version='2.1.92',
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