geneflowdecoder 0.1.1__tar.gz → 0.1.2__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.
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: geneflowdecoder
3
- Version: 0.1.1
3
+ Version: 0.1.2
4
4
  Summary: Gene Transformer + Conditional Rectified Flow decoder from cell-state z to log1p expression X.
5
5
  Author: GeneFlowDecoder Contributors
6
6
  License: MIT
@@ -1,6 +1,6 @@
1
1
  [project]
2
2
  name = "geneflowdecoder"
3
- version = "0.1.1"
3
+ version = "0.1.2"
4
4
  description = "Gene Transformer + Conditional Rectified Flow decoder from cell-state z to log1p expression X."
5
5
  readme = "README.md"
6
6
  requires-python = ">=3.9"
@@ -497,9 +497,9 @@ class GeneFlowDecoder(nn.Module):
497
497
  # 把 .X 或 obsm 留在 cuda 的情况),强制关闭,避免
498
498
  # RuntimeError: cannot pin 'torch.cuda.FloatTensor'
499
499
  # 注:_as_tensor 对 cuda 输入原样返回,因此这里 X/z 仍在 GPU。
500
- assert X.device.type == "cpu" or z.device.type == "cpu" or not pin_memory
501
500
  pin_memory = _want_pin_memory(X, z)
502
-
501
+ assert X.device.type == "cpu" or z.device.type == "cpu" or not pin_memory
502
+
503
503
  ds = torch.utils.data.TensorDataset(X, z)
504
504
  dl = torch.utils.data.DataLoader(
505
505
  ds,
@@ -1,2 +1,2 @@
1
1
  # -*- coding: utf-8 -*-
2
- __version__ = "0.1.1"
2
+ __version__ = "0.1.2"
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: geneflowdecoder
3
- Version: 0.1.1
3
+ Version: 0.1.2
4
4
  Summary: Gene Transformer + Conditional Rectified Flow decoder from cell-state z to log1p expression X.
5
5
  Author: GeneFlowDecoder Contributors
6
6
  License: MIT
File without changes