...

herolib/infra/tmuxrunner/model.py

@@ -1,28 +1,6 @@
-import os
-import time
-import re
-import sys
-import toml
-import libtmux
-from libtmux.pane import Pane
-from libtmux.window import Window
-from libtmux.session import Session
-import psutil
-from typing import Dict, List, Optional, Any, Set, Tuple
-from dataclasses import dataclass, field, asdict
+from typing import Dict, List, Optional
+from dataclasses import dataclass, field
 from datetime import datetime
-import uuid
-from pathlib import Path
-import asyncio
-import uvicorn
-from fastapi import FastAPI, HTTPException
-from fastapi.middleware.cors import CORSMiddleware
-from pydantic import BaseModel
-import threading
-
-# Configuration
-WAITING_MESSAGE = "WAITING FOR JOBS"
-HPY_SH_PATH = "/root/heromonkey/functions/hpy.sh" # Path to hpy.sh
 
 @dataclass
 class ProcessMetrics:
@@ -81,6 +59,6 @@ class DAGStructure:
 
 class MetaData:
     """Class to hold metadata for a task directory."""
-    def __init__(self, timeout: int = 600): # Default timeout to 10 minutes (600 seconds)
+    def __init__(self, timeout: int = 600):  # Default timeout to 10 minutes (600 seconds)
         self.timeout = timeout
-        # Add more attributes here in the future
+        # Add more attributes here in the future