Skip to content

Commit

Permalink
code rabbit fixes
Browse files Browse the repository at this point in the history
  • Loading branch information
emdavis02 committed Jun 24, 2024
1 parent 6f0c929 commit bd2b016
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions sleap/nn/inference.py
Original file line number Diff line number Diff line change
Expand Up @@ -5303,7 +5303,7 @@ def _make_provider_from_cli(args: argparse.Namespace) -> Tuple[Provider, str]:
for file_path in data_path.iterdir():
if file_path.is_file():
data_path_list.append(file_path)
elif data_path.is_file:
elif data_path.is_file():
data_path_list = [args.data_path]
else:
raise ValueError(
Expand Down Expand Up @@ -5353,7 +5353,7 @@ def _make_provider_from_cli(args: argparse.Namespace) -> Tuple[Provider, str]:
print(f"Video: {data_path_file}")
tmp_data_path_list.append(data_path_file)
# TODO: Clean this up.
except Exception as e:
except Exception:
print(f"Error reading file: {data_path_file}")

data_path_list = tmp_data_path_list
Expand Down Expand Up @@ -5531,6 +5531,7 @@ def main(args: Optional[list] = None):

else:
output_path = output_path + "/" + (data_path.stem + ".predictions.slp")
output_path_obj = Path(output_path)

labels_pr.provenance["model_paths"] = predictor.model_paths
labels_pr.provenance["predictor"] = type(predictor).__name__
Expand Down

0 comments on commit bd2b016

Please sign in to comment.