From f3bf77b4161c20d9cd9db2460848ac3ef5d88f7c Mon Sep 17 00:00:00 2001
From: Lakshantha Dissanayake <lakshanthad@yahoo.com>
Date: Fri, 24 May 2024 01:09:37 -0700
Subject: [PATCH] Improve `tflite` exports for Tests CI (#13094)

---
 ultralytics/engine/exporter.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/ultralytics/engine/exporter.py b/ultralytics/engine/exporter.py
index 82675be620..20eb87d2a3 100644
--- a/ultralytics/engine/exporter.py
+++ b/ultralytics/engine/exporter.py
@@ -75,7 +75,6 @@ from ultralytics.utils import (
     ARM64,
     DEFAULT_CFG,
     IS_JETSON,
-    IS_RASPBERRYPI,
     LINUX,
     LOGGER,
     MACOS,
@@ -1012,7 +1011,7 @@ class Exporter:
         """Add metadata to *.tflite models per https://www.tensorflow.org/lite/models/convert/metadata."""
         import flatbuffers
 
-        if MACOS or IS_RASPBERRYPI:  # TODO: should this be 'if ARM64:'?
+        if ARM64:
             from tflite_support import metadata  # noqa
             from tflite_support import metadata_schema_py_generated as schema  # noqa
         else: