From 0178d41bf2d87f666aeed34085236a64bdb7c942 Mon Sep 17 00:00:00 2001
From: Daniel Dowler <12484302+dandawg@users.noreply.github.com>
Date: Mon, 10 Feb 2025 05:42:05 -0700
Subject: [PATCH] chore(backend): fixed imports on metadata_helpers.py (#11604)

Signed-off-by: Daniel Dowler <12484302+dandawg@users.noreply.github.com>
---
 backend/metadata_writer/src/metadata_helpers.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/backend/metadata_writer/src/metadata_helpers.py b/backend/metadata_writer/src/metadata_helpers.py
index e18eeea96e7..5b0937ed4e8 100644
--- a/backend/metadata_writer/src/metadata_helpers.py
+++ b/backend/metadata_writer/src/metadata_helpers.py
@@ -12,10 +12,10 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+import functools
 import json
 import os
 import sys
-import ml_metadata
 from time import sleep
 from ml_metadata.proto import metadata_store_pb2
 from ml_metadata.metadata_store import metadata_store
@@ -169,7 +169,6 @@ def create_context_with_type(
     return context
 
 
-import functools
 @functools.lru_cache(maxsize=128)
 def get_context_by_name(
     store,