fix import

Signed-off-by: Yu Chin Fabian Lim <[email protected]>

tests/acceleration/test_acceleration_framework.py

@@ -17,14 +17,14 @@
 from typing import Annotated
 from unittest.mock import patch
 import copy
+import os
 import tempfile
 
 # Third Party
 import pytest
 import torch
 
 # First Party
-from tests.data import TWITTER_COMPLAINTS_JSON_FORMAT, TWITTER_COMPLAINTS_TOKENIZED
 from tests.test_sft_trainer import DATA_ARGS, MODEL_ARGS, PEFT_LORA_ARGS, TRAIN_ARGS
 
 # Local
@@ -53,6 +53,14 @@
 )
 from tuning.utils.import_utils import is_fms_accelerate_available
 
+TWITTER_COMPLAINTS_JSON_FORMAT = os.path.join(
+    os.path.dirname(__file__), "../data/twitter_complaints_json.json"
+)
+TWITTER_COMPLAINTS_TOKENIZED = os.path.join(
+    os.path.dirname(__file__),
+    "../data/twitter_complaints_tokenized_with_maykeye_tinyllama_v0.json",
+)
+
 # pylint: disable=import-error
 if is_fms_accelerate_available():