SURE-tools 2.0.5__py3-none-any.whl → 2.0.6__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/SURE.py CHANGED
@@ -11,7 +11,7 @@ from torch.distributions import constraints
11
11
  from torch.distributions.transforms import SoftmaxTransform
12
12
 
13
13
  from .utils.custom_mlp import MLP, Exp
14
- from .utils.utils import CustomDataset, CustomDataset3, CustomDataset4, tensor_to_numpy, convert_to_tensor
14
+ from .utils.utils import CustomDataset, CustomDataset2, CustomDataset4, tensor_to_numpy, convert_to_tensor
15
15
 
16
16
 
17
17
  import os
@@ -836,13 +836,14 @@ class SURE(nn.Module):
836
836
  """
837
837
  xs = self.preprocess(xs)
838
838
  xs = convert_to_tensor(xs, device=self.get_device())
839
- dataset = CustomDataset(xs)
839
+ ps = convert_to_tensor(perturb, device=self.get_device())
840
+ dataset = CustomDataset2(xs,ps)
840
841
  dataloader = DataLoader(dataset, batch_size=batch_size, shuffle=False)
841
842
 
842
843
  Z = []
843
844
  with tqdm(total=len(dataloader), desc='', unit='batch') as pbar:
844
- for X_batch, _ in dataloader:
845
- zns = self._cell_move(X_batch, factor_idx, perturb)
845
+ for X_batch, P_batch, _ in dataloader:
846
+ zns = self._cell_move(X_batch, factor_idx, P_batch)
846
847
  Z.append(tensor_to_numpy(zns))
847
848
  pbar.update(1)
848
849
 
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: SURE-tools
3
- Version: 2.0.5
3
+ Version: 2.0.6
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/SURE.py,sha256=0brySH3zXtYAW8eff5BzL1vYq_nmR82wpEPeH-FcLE8,50527
1
+ SURE/SURE.py,sha256=SZHxzI1YXVsdOAIFNE56pH3m_b6-mMegZuUYLnf9t3Y,50606
2
2
  SURE/SURE2.py,sha256=8wlnMwb1xuf9QUksNkWdWx5ZWq-xIy9NLx8RdUnE82o,48501
3
3
  SURE/__init__.py,sha256=mxdLoWa2fPEM4-WkJetMfcecpzqJ8YsM4lOoKHSLTrU,207
4
4
  SURE/assembly/__init__.py,sha256=jxZLURXKPzXe21LhrZ09LgZr33iqdjlQy4oSEj5gR2Q,172
@@ -15,9 +15,9 @@ SURE/utils/__init__.py,sha256=Htqv4KqVKcRiaaTBsR-6yZ4LSlbhbzutjNKXGD9-uds,660
15
15
  SURE/utils/custom_mlp.py,sha256=07TYX1HgxfEjb_3i5MpiZfNhOhx3dKntuwGkrpteWiM,7036
16
16
  SURE/utils/queue.py,sha256=E_5PA5EWcBoGAZj8BkKQnkCK0p4C-4-xcTPqdIXaPXU,1892
17
17
  SURE/utils/utils.py,sha256=IUHjDDtYaAYllCWsZyIzqQwaLul6fJRvHRH4vIYcR-c,8462
18
- sure_tools-2.0.5.dist-info/licenses/LICENSE,sha256=TFHKwmrAViXQbSX5W-NDItkWFjm45HWOeUniDrqmnu0,1065
19
- sure_tools-2.0.5.dist-info/METADATA,sha256=DSDQeffAyyRgBfKxdWTsk7DBd71C3zGdnMlKY4bz1_g,2650
20
- sure_tools-2.0.5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
21
- sure_tools-2.0.5.dist-info/entry_points.txt,sha256=u12payZYgCBy5FCwRHP6AlSQhKCiWSEDwj68r1DVdn8,40
22
- sure_tools-2.0.5.dist-info/top_level.txt,sha256=BtFTebdiJeqra4r6mm-uEtwVRFLZ_IjYsQ7OnalrOvY,5
23
- sure_tools-2.0.5.dist-info/RECORD,,
18
+ sure_tools-2.0.6.dist-info/licenses/LICENSE,sha256=TFHKwmrAViXQbSX5W-NDItkWFjm45HWOeUniDrqmnu0,1065
19
+ sure_tools-2.0.6.dist-info/METADATA,sha256=JxYp73Z3lPkJpz4erR2ugrSGMz4cc98HSs8fhKMByPM,2650
20
+ sure_tools-2.0.6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
21
+ sure_tools-2.0.6.dist-info/entry_points.txt,sha256=u12payZYgCBy5FCwRHP6AlSQhKCiWSEDwj68r1DVdn8,40
22
+ sure_tools-2.0.6.dist-info/top_level.txt,sha256=BtFTebdiJeqra4r6mm-uEtwVRFLZ_IjYsQ7OnalrOvY,5
23
+ sure_tools-2.0.6.dist-info/RECORD,,