Redesign dataset workbench management
This commit is contained in:
@@ -26,14 +26,14 @@ from .coverage import get_coverage_report
|
||||
from .jobs import cancel_job, create_job
|
||||
from .modules.results.service import scan_results, scan_training_curves
|
||||
from .modules.system.service import disk_usage, get_conda_envs, get_gpus, get_runtime_readiness
|
||||
from .modules.dataset.service import create_dataset, generate_yolo_dataset_yaml, list_uploaded_datasets, save_upload, validate_dataset
|
||||
from .modules.dataset.service import copy_dataset, create_dataset, delete_dataset, describe_dataset, generate_yolo_dataset_yaml, list_uploaded_datasets, save_upload, validate_dataset
|
||||
from .modules.weights.service import load_manifest, sync_weights, verify_weights
|
||||
from .agents.evaluation_agent import evaluate_project
|
||||
from .agents.user_agent import latest_user_agent_report, run_user_agent
|
||||
from .agents.validation_agent import validate_project
|
||||
from .paths import ensure_inside
|
||||
from .progress import progress_from_log_path
|
||||
from .schemas import DatasetCreate, DatasetYoloYamlRequest, JobCreate, ProfileCreate, WeightSyncRequest
|
||||
from .schemas import DatasetCopyRequest, DatasetCreate, DatasetYoloYamlRequest, JobCreate, ProfileCreate, WeightSyncRequest
|
||||
|
||||
app = FastAPI(title="Seg Data Server Net", version="0.1.0")
|
||||
|
||||
@@ -149,9 +149,36 @@ def api_datasets() -> list[dict]:
|
||||
return list_uploaded_datasets()
|
||||
|
||||
|
||||
@app.get("/api/datasets/{dataset_name}")
|
||||
def api_dataset(dataset_name: str) -> dict:
|
||||
try:
|
||||
return describe_dataset(dataset_name)
|
||||
except Exception as exc:
|
||||
raise HTTPException(status_code=400, detail=str(exc)) from exc
|
||||
|
||||
|
||||
@app.post("/api/datasets")
|
||||
def api_create_dataset(dataset: DatasetCreate) -> dict:
|
||||
return create_dataset(dataset.name, dataset.description)
|
||||
try:
|
||||
return create_dataset(dataset.name, dataset.description)
|
||||
except Exception as exc:
|
||||
raise HTTPException(status_code=400, detail=str(exc)) from exc
|
||||
|
||||
|
||||
@app.post("/api/datasets/{dataset_name}/copy")
|
||||
def api_copy_dataset(dataset_name: str, request: DatasetCopyRequest) -> dict:
|
||||
try:
|
||||
return copy_dataset(dataset_name, request.name, request.description)
|
||||
except Exception as exc:
|
||||
raise HTTPException(status_code=400, detail=str(exc)) from exc
|
||||
|
||||
|
||||
@app.delete("/api/datasets/{dataset_name}")
|
||||
def api_delete_dataset(dataset_name: str) -> dict:
|
||||
try:
|
||||
return delete_dataset(dataset_name)
|
||||
except Exception as exc:
|
||||
raise HTTPException(status_code=400, detail=str(exc)) from exc
|
||||
|
||||
|
||||
@app.post("/api/datasets/{dataset_name}/upload/{kind}")
|
||||
|
||||
Reference in New Issue
Block a user