diff --git a/scene_checks.py b/scene_checks.py index d1d6971152052e9268828dc01a60b894cb9672df..828ef71b8733dce5f4ece2eadf15eef8018f0f6c 100644 --- a/scene_checks.py +++ b/scene_checks.py @@ -24,7 +24,7 @@ def safing_check(frame, parameters): if not np.in1d(['hatchOpen','sceneMirrorPosition'], frame.columns).all(): return frame - hatch_closing = ((frame.hatchOpen == 1) & ((frame.hatchOpen == -3).diff(-1) == 1)).shift(1) + hatch_closing = (frame.hatchOpen == 1).shift(1) & (frame.hatchOpen == -3) mirror_safing = (hatch_closing & frame.sceneMirrorPosition.isin([ord('H'), ord('A')])) frame['safing_check'] = mirror_safing * 1 annotate_all(frame, mirror_safing, 'mirror likely safed during view')