quotas

type Client struct {
	// contains filtered or unexported fields
}

type QuotaFamily string

type QuotaInfo struct {
	Region string // AWS region this snapshot covers, e.g. "us-east-1"

	// On-Demand quotas — maximum vCPUs per family
	OnDemand map[QuotaFamily]int32

	// Spot quotas — maximum vCPUs per family for Spot instances
	Spot map[QuotaFamily]int32

	// Current usage — vCPUs currently in use (running + pending) per family
	Usage map[QuotaFamily]int32

	RunningInstances     int32     // Current count of running+pending instances in this region
	RunningInstancesMax  int32     // Per-region instance count limit (typically 20 for new accounts)
	LastUpdated          time.Time // When this snapshot was fetched
	CredentialsAvailable bool      // False when quotas were estimated due to missing credentials
}

type SageMakerQuota struct {
	Name  string
	Code  string
	Value float64
}

type SageMakerTypeQuota struct {
	// InstanceType is the ml.*-prefixed type, e.g. "ml.g5.2xlarge".
	InstanceType string
	// TrainingJobLimit is the account limit for "training job usage" (the count
	// of concurrent instances of this type in training jobs). -1 when the region
	// exposes no training-job quota for the type.
	TrainingJobLimit float64
	// ManagedSpotEligible is true when the region exposes a "spot training job
	// usage" quota for the type — i.e. the type can be used with managed spot
	// training. (Managed spot is a billed-time discount, not a spot market, so
	// there is no per-type spot price to report; this only marks eligibility.)
	ManagedSpotEligible bool
}

type ServiceQuotasClient struct {
	// contains filtered or unexported fields
}

type ServiceQuotasLister interface {
	ListSageMakerInstanceQuotas(ctx context.Context, region string) ([]SageMakerQuota, error)
}