From fba77fdebd90167c06553d685b0a07f610b248f8 Mon Sep 17 00:00:00 2001
From: Roman Solomatin <36135455+Samoed@users.noreply.github.com>
Date: Sat, 5 Apr 2025 15:12:50 +0300
Subject: [PATCH] SpeedTask add deprecated warning

---
 mteb/abstasks/AbsTaskSpeedTask.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/mteb/abstasks/AbsTaskSpeedTask.py b/mteb/abstasks/AbsTaskSpeedTask.py
index 0d3f8ae8dd..ff095d8332 100644
--- a/mteb/abstasks/AbsTaskSpeedTask.py
+++ b/mteb/abstasks/AbsTaskSpeedTask.py
@@ -3,6 +3,7 @@
 import logging
 import platform
 import time
+import warnings
 from pathlib import Path
 
 import numpy as np
@@ -85,6 +86,10 @@ def get_system_info(self) -> dict[str, str]:
         return info
 
     def _evaluate_subset(self, model: Encoder, data_split, **kwargs) -> ScoresDict:
+        warnings.warn(
+            "SpeedTask is deprecated and will be removed in `v2`.",
+            DeprecationWarning,
+        )
         model.encode(
             ["encode this"], device=self.device, task_name=self.metadata.name
         )  # ensure model is loaded