Spaces:
Sleeping
Sleeping
Commit ·
81abf49
1
Parent(s): 13f248e
fixed
Browse files- inference.py +0 -4
inference.py
CHANGED
|
@@ -14,7 +14,6 @@ from dotenv import load_dotenv
|
|
| 14 |
load_dotenv()
|
| 15 |
|
| 16 |
from env.environment import SQLDebuggerEnvironment
|
| 17 |
-
from env.models import Action, ActionType, DifficultyLevel
|
| 18 |
# ─────────────────────────────────────────────
|
| 19 |
# ENVIRONMENT VARIABLES
|
| 20 |
# ─────────────────────────────────────────────
|
|
@@ -26,9 +25,6 @@ if HF_TOKEN is None:
|
|
| 26 |
raise ValueError("HF_TOKEN environment variable is required")
|
| 27 |
|
| 28 |
API_KEY = HF_TOKEN
|
| 29 |
-
BENCHMARK = "sql-query-debugger"
|
| 30 |
-
MAX_STEPS = 10
|
| 31 |
-
SUCCESS_SCORE_THRESHOLD = 0.5
|
| 32 |
# ─────────────────────────────────────────────
|
| 33 |
|
| 34 |
def log_start(task: str, env: str, model: str) -> None:
|
|
|
|
| 14 |
load_dotenv()
|
| 15 |
|
| 16 |
from env.environment import SQLDebuggerEnvironment
|
|
|
|
| 17 |
# ─────────────────────────────────────────────
|
| 18 |
# ENVIRONMENT VARIABLES
|
| 19 |
# ─────────────────────────────────────────────
|
|
|
|
| 25 |
raise ValueError("HF_TOKEN environment variable is required")
|
| 26 |
|
| 27 |
API_KEY = HF_TOKEN
|
|
|
|
|
|
|
|
|
|
| 28 |
# ─────────────────────────────────────────────
|
| 29 |
|
| 30 |
def log_start(task: str, env: str, model: str) -> None:
|