From 16c3f8046d102016ed1b5cfab635b04168e5acd2 Mon Sep 17 00:00:00 2001
From: Amirhossein Kazemnejad <2122102+kazemnejad@users.noreply.github.com>
Date: Sun, 31 Dec 2023 14:02:20 -0500
Subject: [PATCH] Allow the test to pass if custom implementation is used

---
 tests/test_task.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/tests/test_task.py b/tests/test_task.py
index 3c1872b..e8ce065 100644
--- a/tests/test_task.py
+++ b/tests/test_task.py
@@ -68,6 +68,10 @@ def test_split_file(task_obj: Task):
 
 def test_task_config_matches_provided_sets(task_obj: Task):
     """Test case to verify if the task config matches the provided sets"""
+    class_dict = task_obj.__class__.__dict__
+    if "get_datasets_raw" in class_dict and "get_prepared_datasets" in class_dict:
+        pytest.skip("Task has custom get_datasets_raw and prepared_dataset builder.")
+
     datasets_raw = task_obj.get_datasets_raw()
 
     task_sets = [DatasetSplit.TEST.value]