mltrail-verifiable-provenan.../mltrail_verifiable_provenan.../registry.py

15 lines
499 B
Python

from typing import Dict, Any
class ContractRegistry:
def __init__(self) -> None:
self._contracts: Dict[str, Dict[str, Any]] = {}
def register_contract(self, name: str, schema: Dict[str, Any], version: str = "1.0.0") -> None:
self._contracts[name] = {"schema": schema, "version": version}
def get_contract(self, name: str) -> Dict[str, Any] | None:
return self._contracts.get(name)
def all_contracts(self) -> Dict[str, Any]:
return self._contracts