Skip to content

Commit 82228bc

Browse files
authored
Merge pull request mala-project#673 from elcorto/feature-fix-grid-dim-type
Fix grid dimension type conversion (mala-project#669)
2 parents 4a46f67 + 79d8acd commit 82228bc

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

mala/network/predictor.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -114,8 +114,8 @@ def predict_for_atoms(self, atoms, gather_ldos=False, temperature=None):
114114
# We only need the diagonal elements.
115115
factor = np.diag(new_cell) / np.diag(old_cell)
116116
factor = factor.astype(int)
117-
self.data.grid_dimension = (
118-
factor * self.data.target_calculator.grid_dimensions
117+
self.data.grid_dimension = list(
118+
map(int, factor * self.data.target_calculator.grid_dimensions)
119119
)
120120

121121
self.data.grid_size = np.prod(self.data.grid_dimension)

0 commit comments

Comments
 (0)