21 lines
896 B
Python
21 lines
896 B
Python
|
import json
|
||
|
|
|
||
|
|
from django.core.management.base import BaseCommand, CommandError
|
||
|
|
|
||
|
|
from control_plane.model_studio.services import ModelStudioService
|
||
|
|
from control_plane.model_studio.models import TrainingProject
|
||
|
|
|
||
|
|
|
||
|
|
class Command(BaseCommand):
|
||
|
|
help = "Classify Guard dataset manifests and record provenance/contamination evidence without changing source data."
|
||
|
|
|
||
|
|
def add_arguments(self, parser):
|
||
|
|
parser.add_argument("--project", required=True, help="TrainingProject slug")
|
||
|
|
|
||
|
|
def handle(self, *args, **options):
|
||
|
|
project = TrainingProject.objects.filter(slug=options["project"]).first()
|
||
|
|
if project is None:
|
||
|
|
raise CommandError("TrainingProject not found.")
|
||
|
|
report = ModelStudioService().curate_guard_datasets(project)
|
||
|
|
self.stdout.write(json.dumps({key: report[key] for key in ["total", "valid", "warning", "blocked", "decision"]}, indent=2))
|