Skip to content
Snippets Groups Projects
Commit a8d7b5d2 authored by Studer Gabriel's avatar Studer Gabriel
Browse files

lDDT: consistent use of ndarrays, also in case of empty default values

parent 103c2c92
Branches
Tags
No related merge requests found
...@@ -841,10 +841,10 @@ class lDDTScorer: ...@@ -841,10 +841,10 @@ class lDDTScorer:
"""Compute distance related members of lDDTScorer """Compute distance related members of lDDTScorer
""" """
# init # init
self._ref_indices = [[] for idx in range(self.n_atoms)] self._ref_indices = [np.asarray([], dtype=np.int64) for idx in range(self.n_atoms)]
self._ref_distances = [[] for idx in range(self.n_atoms)] self._ref_distances = [np.asarray([], dtype=np.float64) for idx in range(self.n_atoms)]
self._sym_ref_indices = [[] for idx in range(self.n_atoms)] self._sym_ref_indices = [np.asarray([], dtype=np.int64) for idx in range(self.n_atoms)]
self._sym_ref_distances = [[] for idx in range(self.n_atoms)] self._sym_ref_distances = [np.asarray([], dtype=np.float64) for idx in range(self.n_atoms)]
# initialize positions with values far in nirvana. If a position is not # initialize positions with values far in nirvana. If a position is not
# set, it should be far away from any position in target (or at least # set, it should be far away from any position in target (or at least
...@@ -888,10 +888,10 @@ class lDDTScorer: ...@@ -888,10 +888,10 @@ class lDDTScorer:
"""Select subset of contacts only covering intra-chain contacts """Select subset of contacts only covering intra-chain contacts
""" """
# init # init
self._ref_indices_sc = [[] for idx in range(self.n_atoms)] self._ref_indices_sc = [np.asarray([], dtype=np.int64) for idx in range(self.n_atoms)]
self._ref_distances_sc = [[] for idx in range(self.n_atoms)] self._ref_distances_sc = [np.asarray([], dtype=np.float64) for idx in range(self.n_atoms)]
self._sym_ref_indices_sc = [[] for idx in range(self.n_atoms)] self._sym_ref_indices_sc = [np.asarray([], dtype=np.int64) for idx in range(self.n_atoms)]
self._sym_ref_distances_sc = [[] for idx in range(self.n_atoms)] self._sym_ref_distances_sc = [np.asarray([], dtype=np.float64) for idx in range(self.n_atoms)]
# start from overall contacts # start from overall contacts
ref_indices = self.ref_indices ref_indices = self.ref_indices
...@@ -920,10 +920,10 @@ class lDDTScorer: ...@@ -920,10 +920,10 @@ class lDDTScorer:
"""Select subset of contacts only covering inter-chain contacts """Select subset of contacts only covering inter-chain contacts
""" """
# init # init
self._ref_indices_ic = [[] for idx in range(self.n_atoms)] self._ref_indices_ic = [np.asarray([], dtype=np.int64) for idx in range(self.n_atoms)]
self._ref_distances_ic = [[] for idx in range(self.n_atoms)] self._ref_distances_ic = [np.asarray([], dtype=np.float64) for idx in range(self.n_atoms)]
self._sym_ref_indices_ic = [[] for idx in range(self.n_atoms)] self._sym_ref_indices_ic = [np.asarray([], dtype=np.int64) for idx in range(self.n_atoms)]
self._sym_ref_distances_ic = [[] for idx in range(self.n_atoms)] self._sym_ref_distances_ic = [np.asarray([], dtype=np.float64) for idx in range(self.n_atoms)]
# start from overall contacts # start from overall contacts
ref_indices = self.ref_indices ref_indices = self.ref_indices
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment