Skip to content

Commit 79d8acd

Browse files
committed
Fix grid dimension type conversion (mala-project#669)
1 parent d65fdf7 commit 79d8acd

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

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)