Adjusted type hint for load_and_concat_data

This commit is contained in:
Sebastian Kloth 2023-11-07 11:28:12 +01:00
parent 1a8134c002
commit a75091aa95

View File

@ -1,6 +1,7 @@
import abc import abc
from dataclasses import dataclass, field from dataclasses import dataclass, field
from subprocess import run from subprocess import run
from typing import Iterable
import pandas as pd import pandas as pd
@ -44,7 +45,7 @@ class MDSystem(abc.ABC):
run(["mv", hdf5_temp_file, hdf5_file]) run(["mv", hdf5_temp_file, hdf5_file])
def load_and_concat_data(systems: list[MDSystem], key: str, verbose: bool = False): def load_and_concat_data(systems: Iterable[MDSystem], key: str, verbose: bool = False):
data = [] data = []
for system in systems: for system in systems:
try: try: