diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml index 9903428..3bbd5e4 100644 --- a/.github/workflows/test.yml +++ b/.github/workflows/test.yml @@ -166,6 +166,9 @@ jobs: with: name: video-processor-coverage path: ./coverage.xml + + - name: Commit formatted changes + run: ./scripts/commit-and-push.sh "[CI] update test" test-engine-and-interface: name: Test Engine and Interface diff --git a/tests/video_processor/steps/test_detection_3d.py b/tests/video_processor/steps/test_detection_3d.py index 11a2dd3..9633232 100644 --- a/tests/video_processor/steps/test_detection_3d.py +++ b/tests/video_processor/steps/test_detection_3d.py @@ -47,10 +47,10 @@ def test_detection_3d(): det_result = FromDetection2DAndRoad.get(output) assert det_result is not None - # with open(os.path.join(OUTPUT_DIR, f'FromDetection2DAndRoad--{name}.json'), 'w') as f: - # json.dump([(d[0].cpu().numpy().tolist(), d[1], d[2]) for d in det_result], f, indent=1) - # with open(os.path.join(OUTPUT_DIR, f'FromDetection2DAndRoad--{name}.pkl'), 'wb') as f: - # pickle.dump([(d[0].cpu(), d[1], d[2]) for d in det_result], f) + with open(os.path.join(OUTPUT_DIR, f'FromDetection2DAndRoad--{name}.json'), 'w') as f: + json.dump([(d[0].cpu().numpy().tolist(), d[1], d[2]) for d in det_result], f, indent=1) + with open(os.path.join(OUTPUT_DIR, f'FromDetection2DAndRoad--{name}.pkl'), 'wb') as f: + pickle.dump([(d[0].cpu(), d[1], d[2]) for d in det_result], f) with open(os.path.join(OUTPUT_DIR, f'FromDetection2DAndRoad--{name}.pkl'), 'rb') as f: det_groundtruth = pickle.load(f)