Add tests before pushing to Docker Hub (#3924)

2023-07-24 23:56:16 +02:00
parent 86b3c001c4
commit 2ee147838a
7 changed files with 71 additions and 69 deletions
--- a/ultralytics/engine/exporter.py
+++ b/ultralytics/engine/exporter.py
@ -465,7 +465,7 @@ class Exporter:
    @try_export
    def export_coreml(self, prefix=colorstr('CoreML:')):
        """YOLOv8 CoreML export."""
-        check_requirements('coremltools>=6.0')
+        check_requirements('coremltools>=6.0,<=6.2')
        import coremltools as ct  # noqa

        LOGGER.info(f'\n{prefix} starting export with coremltools {ct.__version__}...')