From 4ce9d266d86c9fba6b17b88db1550ffd194c0252 Mon Sep 17 00:00:00 2001
From: Alexander Myasoedov <msoedov@gmail.com>
Date: Tue, 10 Dec 2024 19:00:48 +0200
Subject: [PATCH] fix(linter):

---
 agentic_security/probe_data/msj_data.py      | 8 --------
 agentic_security/probe_data/test_msj_data.py | 2 +-
 2 files changed, 1 insertion(+), 9 deletions(-)

diff --git a/agentic_security/probe_data/msj_data.py b/agentic_security/probe_data/msj_data.py
index 6861851..f64c4aa 100644
--- a/agentic_security/probe_data/msj_data.py
+++ b/agentic_security/probe_data/msj_data.py
@@ -1,13 +1,6 @@
-import io
-import os
-import random
 from dataclasses import dataclass
-from functools import lru_cache
 
-import httpx
-import pandas as pd
 from cache_to_disk import cache_to_disk
-from loguru import logger
 
 
 # TODO: refactor this class to use from .data
@@ -47,7 +40,6 @@ def load_dataset_generic(name, getter=lambda x: x["train"]["prompt"]):
 def prepare_prompts(
     dataset_names=[], budget=-1, tools_inbox=None
 ) -> list[ProbeDataset]:
-
     # fka/awesome-chatgpt-prompts
     # data-is-better-together/10k_prompts_ranked
     # alespalla/chatbot_instruction_prompts
diff --git a/agentic_security/probe_data/test_msj_data.py b/agentic_security/probe_data/test_msj_data.py
index f5ee6ab..e81812b 100644
--- a/agentic_security/probe_data/test_msj_data.py
+++ b/agentic_security/probe_data/test_msj_data.py
@@ -1,4 +1,4 @@
-from unittest.mock import patch, Mock
+from unittest.mock import patch
 
 from agentic_security.probe_data.msj_data import (
     ProbeDataset,