Révision e1c5f171

b/modules/formats.py
183 183
    atoms_list = []
184 184
    for conf in conf_list:
185 185
        if code == 'cp2k':
186
            read_coords_cp2k(glob(f"{run_type}/{conf}/*-1.restart")[0],
187
                             spec_atms)
186
            atoms_list.append(read_coords_cp2k(glob(f"{run_type}/{conf}/*-1"
187
                                                    f".restart")[0], spec_atms))
188 188
        # elif code == 'vasp'
189 189
    return atoms_list
190 190

  
b/modules/screening.py
9 9
    atoms.info[prop_name] = prop_val
10 10

  
11 11

  
12
def select_confs(orig_conf_list: list, calc_dirs: list, magns: list, num_sel: int, code: str):
12
def select_confs(orig_conf_list: list, calc_dirs: list, magns: list,
13
                 num_sel: int, code: str):
13 14
    """Takes a list ase.Atoms and selects the most different magnitude-wise.
14 15

  
15 16
    Given a list of ase.Atoms objects and a list of magnitudes, it selects a
......
29 30
    from modules.formats import collect_energies
30 31

  
31 32
    conf_list = deepcopy(orig_conf_list)
32
    selected_ids = []
33
    conf_enrgs, mois, selected_ids = [], [], []
33 34
    if num_sel >= len(conf_list):
34 35
        logger.warning('Number of conformers per magnitude is equal or larger '
35 36
                       'than the total number of conformers. Using all '
......
65 66
    return [conf_list[idx] for idx in selected_ids]
66 67

  
67 68

  
68
def get_vect_angle(v1, v2, degrees=False):
69
def get_vect_angle(v1: list, v2: list, degrees=False):
69 70
    """Computes the angle between two vectors.
70 71

  
71 72
    @param v1: The first vector.

Formats disponibles : Unified diff