cSAXS/cdtools
0
0
Fork 0
mirror of https://github.com/cdtools-developers/cdtools.git synced 2026-09-09 21:12:42 +02:00
Code Issues Packages Projects Releases Wiki Activity

Remove last of t.tensor usage from dataset loading

Browse Source
This commit is contained in:
Abe Levitan
2019-10-04 17:00:43 -04:00
parent 120e23ef15
commit e987d40df8
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
CDTools/datasets/ptycho_2d_dataset.py
+2 -2
View File
@@ -62,8 +62,8 @@ class Ptycho2DDataset(CDataset):
super(Ptycho2DDataset,self).__init__(*args, **kwargs)
self.axes = copy(axes)
self.translations = t.tensor(translations)
self.patterns = t.tensor(patterns)
self.translations = t.Tensor(translations).clone()
self.patterns = t.Tensor(patterns).clone()
if self.mask is None:
self.mask = t.ones(self.patterns.shape[-2:]).to(dtype=t.bool)
self.mask.masked_fill_(t.isnan(t.sum(self.patterns,dim=(0,))),0)
tests/test_datasets.py
+2 -2
View File
@@ -152,8 +152,8 @@ def test_Ptycho2DDataset_init():
assert dataset.sample_info == sample_info
assert dataset.wavelength == wavelength
assert dataset.detector_geometry == detector_geometry
assert t.allclose(dataset.patterns, t.tensor(patterns))
assert t.allclose(dataset.translations, t.tensor(translations))
assert t.allclose(dataset.patterns, t.Tensor(patterns))
assert t.allclose(dataset.translations, t.Tensor(translations))