SURE-tools 2.0.5__tar.gz → 2.0.6__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.
- {sure_tools-2.0.5 → sure_tools-2.0.6}/PKG-INFO +1 -1
- {sure_tools-2.0.5 → sure_tools-2.0.6}/SURE/SURE.py +5 -4
- {sure_tools-2.0.5 → sure_tools-2.0.6}/SURE_tools.egg-info/PKG-INFO +1 -1
- {sure_tools-2.0.5 → sure_tools-2.0.6}/setup.py +1 -1
- {sure_tools-2.0.5 → sure_tools-2.0.6}/LICENSE +0 -0
- {sure_tools-2.0.5 → sure_tools-2.0.6}/README.md +0 -0
- {sure_tools-2.0.5 → sure_tools-2.0.6}/SURE/__init__.py +0 -0
- {sure_tools-2.0.5 → sure_tools-2.0.6}/SURE/assembly/__init__.py +0 -0
- {sure_tools-2.0.5 → sure_tools-2.0.6}/SURE/assembly/assembly.py +0 -0
- {sure_tools-2.0.5 → sure_tools-2.0.6}/SURE/assembly/atlas.py +0 -0
- {sure_tools-2.0.5 → sure_tools-2.0.6}/SURE/atac/__init__.py +0 -0
- {sure_tools-2.0.5 → sure_tools-2.0.6}/SURE/atac/utils.py +0 -0
- {sure_tools-2.0.5 → sure_tools-2.0.6}/SURE/codebook/__init__.py +0 -0
- {sure_tools-2.0.5 → sure_tools-2.0.6}/SURE/codebook/codebook.py +0 -0
- {sure_tools-2.0.5 → sure_tools-2.0.6}/SURE/flow/__init__.py +0 -0
- {sure_tools-2.0.5 → sure_tools-2.0.6}/SURE/flow/plot_quiver.py +0 -0
- {sure_tools-2.0.5 → sure_tools-2.0.6}/SURE/utils/__init__.py +0 -0
- {sure_tools-2.0.5 → sure_tools-2.0.6}/SURE/utils/custom_mlp.py +0 -0
- {sure_tools-2.0.5 → sure_tools-2.0.6}/SURE/utils/queue.py +0 -0
- {sure_tools-2.0.5 → sure_tools-2.0.6}/SURE/utils/utils.py +0 -0
- {sure_tools-2.0.5 → sure_tools-2.0.6}/SURE_tools.egg-info/SOURCES.txt +0 -0
- {sure_tools-2.0.5 → sure_tools-2.0.6}/SURE_tools.egg-info/dependency_links.txt +0 -0
- {sure_tools-2.0.5 → sure_tools-2.0.6}/SURE_tools.egg-info/entry_points.txt +0 -0
- {sure_tools-2.0.5 → sure_tools-2.0.6}/SURE_tools.egg-info/requires.txt +0 -0
- {sure_tools-2.0.5 → sure_tools-2.0.6}/SURE_tools.egg-info/top_level.txt +0 -0
- {sure_tools-2.0.5 → sure_tools-2.0.6}/setup.cfg +0 -0
|
@@ -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,
|
|
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
|
-
|
|
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,
|
|
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
|
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|