From 26943d9c6228b1d80d122add8f365072ea558a33 Mon Sep 17 00:00:00 2001 From: Jonathan Vandermause Date: Sun, 15 Sep 2024 08:36:34 -0400 Subject: [PATCH] replace ints with int8s --- flare/utils/parameter_helper.py | 2 +- flare/utils/parameters.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/flare/utils/parameter_helper.py b/flare/utils/parameter_helper.py index 3b3aaf8fb..d3ca17fec 100644 --- a/flare/utils/parameter_helper.py +++ b/flare/utils/parameter_helper.py @@ -839,7 +839,7 @@ def summarize_group(self, group_type): for ele in self.groups["specie"][idt]: atom_n = atomic_numbers[ele] if atom_n >= len(self.species_mask): - new_mask = np.ones(atom_n, dtype=np.int) * (nspecie - 1) + new_mask = np.ones(atom_n, dtype=np.int8) * (nspecie - 1) new_mask[: len(self.species_mask)] = self.species_mask self.species_mask = new_mask self.species_mask[atom_n] = idt diff --git a/flare/utils/parameters.py b/flare/utils/parameters.py index 5ed1478fc..7fabd856c 100644 --- a/flare/utils/parameters.py +++ b/flare/utils/parameters.py @@ -196,7 +196,7 @@ def check_instantiation(hyps, cutoffs, kernels, param_dict): # check mask has the right dimension and values mask = param_dict[f"{kernel}_mask"] - param_dict[f"{kernel}_mask"] = nparray(mask, dtype=np.int) + param_dict[f"{kernel}_mask"] = nparray(mask, dtype=np.int8) assert npmax(mask) < n dim = Parameters.ndim[kernel]