From e987d40df83508feaa5de998e82fca48ed986842 Mon Sep 17 00:00:00 2001 From: Abe Levitan Date: Fri, 4 Oct 2019 17:00:43 -0400 Subject: [PATCH] Remove last of t.tensor usage from dataset loading --- CDTools/datasets/ptycho_2d_dataset.py | 4 ++-- tests/test_datasets.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/CDTools/datasets/ptycho_2d_dataset.py b/CDTools/datasets/ptycho_2d_dataset.py index 651893a..c4d417f 100644 --- a/CDTools/datasets/ptycho_2d_dataset.py +++ b/CDTools/datasets/ptycho_2d_dataset.py @@ -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) diff --git a/tests/test_datasets.py b/tests/test_datasets.py index c83e204..7347ad7 100644 --- a/tests/test_datasets.py +++ b/tests/test_datasets.py @@ -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))