✅ [Pass] tests, skip drawing if graphviz not found

WongKinYiu / YOLO

An MIT rewrite of YOLOv9

MIT License

292 stars 19 forks source link

✅ [Pass] tests, skip drawing if graphviz not found #21

Closed LaiEthanLai closed 1 month ago

LaiEthanLai commented 1 month ago

Codes to circumvent draw_model() when the graphviz backend can not be found added. This happens when performing pytest.

[x] The code follows the Python style guide.
[x] Code and files are well organized.
[x] All tests pass.
[x] New code is covered by tests.
[x] We would be very happy if gitmoji😆 could be used to assist the commit message💬!

henrytsui000 commented 1 month ago

Hi Ethan,

Thank you for your contribution! Your changes look great, and I appreciate the addition of the condition to skip drawing if Graphviz is not found. This will definitely help in environments where Graphviz isn't available.

I have reviewed the commit, and everything seems to be in order. All tests have passed as well, which is great. I'll go ahead and merge this PR.

Thanks again for your efforts!

Best regards, Henry Tsui