Skip to content

Commit

Permalink
fix typing too
Browse files Browse the repository at this point in the history
  • Loading branch information
alinelena committed Nov 14, 2024
1 parent 88376ba commit ef21439
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions mace/calculators/mace.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@
import logging
from glob import glob
from pathlib import Path
from typing import Union
from typing import Union, List

import numpy as np
import torch
Expand Down Expand Up @@ -51,7 +51,7 @@ class MACECalculator(Calculator):
def __init__(
self,
model_paths: Union[list, str, None] = None,
models: Union[list[torch.nn.Module], torch.nn.Module, None] = None,
models: Union[List[torch.nn.Module], torch.nn.Module, None] = None,
device: str = "cpu",
energy_units_to_eV: float = 1.0,
length_units_to_A: float = 1.0,
Expand Down

0 comments on commit ef21439

Please sign in to comment.