Skip to content

Commit

Permalink
rolled back np.int64 changes to match v0.2 branch
Browse files Browse the repository at this point in the history
  • Loading branch information
loodvn committed Aug 2, 2023
1 parent 6463ad9 commit 9044bb2
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions evcouplings/compare/pdb.py
Original file line number Diff line number Diff line change
Expand Up @@ -422,9 +422,9 @@ def _get_range(object_counts):
"""
Extract index ranges for chains, residues and atoms
"""
last_element = np.cumsum(object_counts, dtype=np.int64)
last_element = np.cumsum(object_counts, dtype=int)
first_element = np.concatenate(
(np.zeros(1, dtype=np.int64), last_element[:-1])
(np.zeros(1, dtype=int), last_element[:-1])
)

return first_element, last_element
Expand Down Expand Up @@ -576,7 +576,7 @@ def get_chain(self, chain, model=0):
)

# indices of chains that match chain name, in current model
indices = np.arange(first_chain_index, last_chain_index, dtype=np.int64)
indices = np.arange(first_chain_index, last_chain_index, dtype=int)
target_chain_indeces = indices[chain_names == chain]

if len(target_chain_indeces) == 0:
Expand All @@ -598,7 +598,7 @@ def get_chain(self, chain, model=0):
chain_indeces = np.concatenate([
np.full(
self.last_residue_index[i] - self.first_residue_index[i],
i, dtype=np.int64
i, dtype=int
) for i in target_chain_indeces
])

Expand Down

0 comments on commit 9044bb2

Please sign in to comment.