Skip to content

Commit 17d97ae

Browse files
committed
Don't fail export on mcap read error
1 parent 57cd20e commit 17d97ae

1 file changed

Lines changed: 10 additions & 3 deletions

File tree

  • tools/machine-learning/fall_detection/data_loading

tools/machine-learning/fall_detection/data_loading/load_data.py

Lines changed: 10 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -124,9 +124,16 @@ def read_mcap(mcap_path: Path) -> pl.DataFrame:
124124

125125

126126
def convert_mcaps(mcaps: list[str]) -> pl.DataFrame:
127-
return pl.concat(
128-
(read_mcap(Path(mcap)) for mcap in tqdm(mcaps)), how="diagonal"
129-
)
127+
mcap_dataframes = []
128+
for mcap in tqdm(mcaps):
129+
try:
130+
dataframe = read_mcap(Path(mcap))
131+
except:
132+
print(f"Skipping mcap: {mcap}")
133+
134+
mcap_dataframes.append(dataframe)
135+
136+
return pl.concat(mcap_dataframes, how="diagonal")
130137

131138

132139
def load(path: str):

0 commit comments

Comments
 (0)