fix(rm IS_VERCEL):

This commit is contained in:
Alexander Myasoedov
2024-11-30 12:53:34 +02:00
parent 7b086242a3
commit 12bd95b74d
3 changed files with 2 additions and 31 deletions
+1 -15
View File
@@ -6,6 +6,7 @@ from functools import lru_cache
import httpx
import pandas as pd
from cache_to_disk import cache_to_disk
from loguru import logger
from agentic_security.probe_data import stenography_fn
@@ -15,21 +16,6 @@ from agentic_security.probe_data.modules import (
inspect_ai_tool,
)
IS_VERCEL = os.getenv("IS_VERCEL", "f") == "t"
if not IS_VERCEL:
from cache_to_disk import cache_to_disk
else:
# Read only fs in vercel, just mock no-op decorator
def cache_to_disk(*_):
def decorator(fn):
def wrapper(*args, **kwargs):
return fn(*args, **kwargs)
return wrapper
return decorator
@dataclass
class ProbeDataset: