From a2a514dec8a7a96c4442f50885a46abdb4b7fba1 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Fri, 23 Apr 2021 23:50:02 +0200
Subject: [PATCH] Update export.py (#2909)

---
 models/export.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/models/export.py b/models/export.py
index 312e949f56ac..5b7b6bda08ae 100644
--- a/models/export.py
+++ b/models/export.py
@@ -112,7 +112,7 @@
     try:
         import coremltools as ct
 
-        print(f'{prefix} starting export with coremltools {onnx.__version__}...')
+        print(f'{prefix} starting export with coremltools {ct.__version__}...')
         # convert model from torchscript and apply pixel scaling as per detect.py
         model = ct.convert(ts, inputs=[ct.ImageType(name='image', shape=img.shape, scale=1 / 255.0, bias=[0, 0, 0])])
         f = opt.weights.replace('.pt', '.mlmodel')  # filename