From da63d3b2e3c832315c24df575cca12ab076cba0b Mon Sep 17 00:00:00 2001
From: saipraneeth <2506664+msaipraneeth@users.noreply.github.com>
Date: Tue, 12 Dec 2023 07:20:29 +0000
Subject: [PATCH] fix mypy issues

---
 tests/test_utils_build_entities_from_data.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/tests/test_utils_build_entities_from_data.py b/tests/test_utils_build_entities_from_data.py
index e4b942c..774e7e8 100644
--- a/tests/test_utils_build_entities_from_data.py
+++ b/tests/test_utils_build_entities_from_data.py
@@ -1,7 +1,8 @@
+"""Tests for `utils.build_entities_from_data`"""
 import json
 
-from dataintegration.entity import EntitySchema, EntityPath
-from dataintegration.utils import build_entities_from_data
+from cmem_plugin_base.dataintegration.entity import EntitySchema, EntityPath
+from cmem_plugin_base.dataintegration.utils import build_entities_from_data
 
 
 def test_single_object():