Skip to content

Commit

Permalink
lint fix finetuning select
Browse files Browse the repository at this point in the history
  • Loading branch information
ilyes319 committed Jun 19, 2024
1 parent c00d6d7 commit 1cd8189
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions mace/cli/fine_tuning_select.py
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@
try:
import fpsample
except ImportError:
logging.error("fpsample not found, to use FPS, install using pip install fpsample")
pass


def parse_args() -> argparse.Namespace:
Expand Down Expand Up @@ -110,7 +110,7 @@ def calculate_descriptors(


def filter_atoms(
atoms: ase.Atoms, element_subset: list[str], filtering_type: str
atoms: ase.Atoms, element_subset: List[str], filtering_type: str
) -> bool:
"""
Filters atoms based on the provided filtering type and element subset.
Expand Down Expand Up @@ -149,7 +149,7 @@ def filter_atoms(


class FPS:
def __init__(self, atoms_list: list[ase.Atoms], n_samples: int):
def __init__(self, atoms_list: List[ase.Atoms], n_samples: int):
self.n_samples = n_samples
self.atoms_list = atoms_list
self.species = np.unique([x.symbol for atoms in atoms_list for x in atoms])
Expand All @@ -160,7 +160,7 @@ def __init__(self, atoms_list: list[ase.Atoms], n_samples: int):

def run(
self,
) -> list[int]:
) -> List[int]:
"""
Run the farthest point sampling algorithm.
"""
Expand Down

0 comments on commit 1cd8189

Please sign in to comment.