SURE-tools 2.1.82__tar.gz → 2.1.84__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.82 → sure_tools-2.1.84}/PKG-INFO +1 -1
  2. {sure_tools-2.1.82 → sure_tools-2.1.84}/SURE/PerturbFlow.py +3 -0
  3. {sure_tools-2.1.82 → sure_tools-2.1.84}/SURE/utils/custom_mlp.py +4 -1
  4. {sure_tools-2.1.82 → sure_tools-2.1.84}/SURE_tools.egg-info/PKG-INFO +1 -1
  5. {sure_tools-2.1.82 → sure_tools-2.1.84}/setup.py +1 -1
  6. {sure_tools-2.1.82 → sure_tools-2.1.84}/LICENSE +0 -0
  7. {sure_tools-2.1.82 → sure_tools-2.1.84}/README.md +0 -0
  8. {sure_tools-2.1.82 → sure_tools-2.1.84}/SURE/SURE.py +0 -0
  9. {sure_tools-2.1.82 → sure_tools-2.1.84}/SURE/__init__.py +0 -0
  10. {sure_tools-2.1.82 → sure_tools-2.1.84}/SURE/assembly/__init__.py +0 -0
  11. {sure_tools-2.1.82 → sure_tools-2.1.84}/SURE/assembly/assembly.py +0 -0
  12. {sure_tools-2.1.82 → sure_tools-2.1.84}/SURE/assembly/atlas.py +0 -0
  13. {sure_tools-2.1.82 → sure_tools-2.1.84}/SURE/atac/__init__.py +0 -0
  14. {sure_tools-2.1.82 → sure_tools-2.1.84}/SURE/atac/utils.py +0 -0
  15. {sure_tools-2.1.82 → sure_tools-2.1.84}/SURE/codebook/__init__.py +0 -0
  16. {sure_tools-2.1.82 → sure_tools-2.1.84}/SURE/codebook/codebook.py +0 -0
  17. {sure_tools-2.1.82 → sure_tools-2.1.84}/SURE/flow/__init__.py +0 -0
  18. {sure_tools-2.1.82 → sure_tools-2.1.84}/SURE/flow/flow_stats.py +0 -0
  19. {sure_tools-2.1.82 → sure_tools-2.1.84}/SURE/flow/plot_quiver.py +0 -0
  20. {sure_tools-2.1.82 → sure_tools-2.1.84}/SURE/perturb/__init__.py +0 -0
  21. {sure_tools-2.1.82 → sure_tools-2.1.84}/SURE/perturb/perturb.py +0 -0
  22. {sure_tools-2.1.82 → sure_tools-2.1.84}/SURE/utils/__init__.py +0 -0
  23. {sure_tools-2.1.82 → sure_tools-2.1.84}/SURE/utils/queue.py +0 -0
  24. {sure_tools-2.1.82 → sure_tools-2.1.84}/SURE/utils/utils.py +0 -0
  25. {sure_tools-2.1.82 → sure_tools-2.1.84}/SURE_tools.egg-info/SOURCES.txt +0 -0
  26. {sure_tools-2.1.82 → sure_tools-2.1.84}/SURE_tools.egg-info/dependency_links.txt +0 -0
  27. {sure_tools-2.1.82 → sure_tools-2.1.84}/SURE_tools.egg-info/entry_points.txt +0 -0
  28. {sure_tools-2.1.82 → sure_tools-2.1.84}/SURE_tools.egg-info/requires.txt +0 -0
  29. {sure_tools-2.1.82 → sure_tools-2.1.84}/SURE_tools.egg-info/top_level.txt +0 -0
  30. {sure_tools-2.1.82 → sure_tools-2.1.84}/setup.cfg +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: SURE-tools
3
- Version: 2.1.82
3
+ Version: 2.1.84
4
4
  Summary: Succinct Representation of Single Cells
5
5
  Home-page: https://github.com/ZengFLab/SURE
6
6
  Author: Feng Zeng
@@ -681,6 +681,9 @@ class PerturbFlow(nn.Module):
681
681
  zus = zus + self.cell_factor_effect[i]([zns,us[:,i].reshape(-1,1)])
682
682
  return zus
683
683
 
684
+ def _get_codebook_identity(self):
685
+ return torch.eye(self.code_size, **self.options)
686
+
684
687
  def _get_codebook(self):
685
688
  I = torch.eye(self.code_size, **self.options)
686
689
  if self.latent_dist=='studentt':
@@ -239,7 +239,10 @@ class ZeroBiasMLP(nn.Module):
239
239
  def forward(self, x):
240
240
  y = self.mlp(x)
241
241
  mask = torch.zeros_like(y)
242
- mask[x[1][:,0]>0,:] = 1
242
+ if len(y.shape)==2:
243
+ mask[x[1][:,0]>0,:] = 1
244
+ elif len(y.shape)==3:
245
+ mask[:,x[1][:,0]>0,:] = 1
243
246
  return y*mask
244
247
 
245
248
 
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: SURE-tools
3
- Version: 2.1.82
3
+ Version: 2.1.84
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.82',
8
+ version='2.1.84',
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