From 20dd11d586b766f3984fa1b0c7a46ebc14793ae6 Mon Sep 17 00:00:00 2001 From: Francesco Andreuzzi Date: Tue, 6 Dec 2022 11:35:37 +0100 Subject: [PATCH] Select variables to be used as labels in `_Nd_sampler` (#47) * select labels variables --- pina/span.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/pina/span.py b/pina/span.py index cab9544..abb107e 100644 --- a/pina/span.py +++ b/pina/span.py @@ -84,12 +84,12 @@ class Span(Location): def _Nd_sampler(n, mode, variables): """ Sample ll the variables together """ - bounds = torch.tensor( - [v for k, v in self.range_.items() if k in variables] - ) + pairs = [(k, v) for k, v in self.range_.items() if k in variables] + keys, values = map(list, zip(*pairs)) + bounds = torch.tensor(values) result = self._sample_range(n, mode, bounds) result = result.as_subclass(LabelTensor) - result.labels = list(self.range_.keys()) + result.labels = keys for variable in variables: if variable in self.fixed_.keys():