config

type ArgumentType string

type Build struct {
	GPU                bool      `json:"gpu,omitempty" yaml:"gpu,omitempty"`
	PythonVersion      string    `json:"python_version,omitempty" yaml:"python_version"`
	PythonRequirements string    `json:"python_requirements,omitempty" yaml:"python_requirements,omitempty"`
	PythonPackages     []string  `json:"python_packages,omitempty" yaml:"python_packages,omitempty"` // Deprecated, but included for backwards compatibility
	Run                []RunItem `json:"run,omitempty" yaml:"run,omitempty"`
	SystemPackages     []string  `json:"system_packages,omitempty" yaml:"system_packages,omitempty"`
	PreInstall         []string  `json:"pre_install,omitempty" yaml:"pre_install,omitempty"` // Deprecated, but included for backwards compatibility
	CUDA               string    `json:"cuda,omitempty" yaml:"cuda,omitempty"`
	CuDNN              string    `json:"cudnn,omitempty" yaml:"cudnn,omitempty"`
	// SDKVersion pins the cog Python SDK version installed in the container.
	// Accepts a PEP 440 version string (e.g. "0.18.0" or "0.18.0a1").
	// When empty the latest release is installed. Overridden by COG_SDK_WHEEL env var.
	SDKVersion string `json:"sdk_version,omitempty" yaml:"sdk_version,omitempty"`
	// contains filtered or unexported fields
}

type BuildOptions struct {
	// SourceEpochTimestamp is the number of seconds since Unix epoch to use
	// for the build timestamp. Set to -1 to disable timestamp rewrites.
	// This is useful for reproducible builds.
	SourceEpochTimestamp int64

	// XCachePath is the path to the BuildKit cache directory.
	// If empty, inline caching is used instead of local cache.
	XCachePath string
}

type CUDABaseImage struct {
	Tag     string
	CUDA    string
	CuDNN   string
	IsDevel bool
	Ubuntu  string
}

type CompatibilityError struct {
	Component1 string
	Version1   string
	Component2 string
	Version2   string
	Message    string
}

type Concurrency struct {
	Max int `json:"max,omitempty" yaml:"max"`
}

type Config struct {
	Build       *Build         `json:"build" yaml:"build"`
	Image       string         `json:"image,omitempty" yaml:"image,omitempty"`
	Predict     string         `json:"predict,omitempty" yaml:"predict"`
	Train       string         `json:"train,omitempty" yaml:"train,omitempty"`
	Concurrency *Concurrency   `json:"concurrency,omitempty" yaml:"concurrency,omitempty"`
	Environment []string       `json:"environment,omitempty" yaml:"environment,omitempty"`
	Weights     []WeightSource `json:"weights,omitempty" yaml:"weights,omitempty"`
	// contains filtered or unexported fields
}

type ConfigError interface {
	error
	ConfigError() // marker method
}

type DeprecationWarning struct {
	Field       string
	Replacement string
	Message     string
}

type LoadResult struct {
	Config   *Config
	Warnings []DeprecationWarning
	RootDir  string
}

type ParseError struct {
	Filename string
	Err      error
}

type RunArgument struct {
	Type    ArgumentType `json:"type"`
	Default *string      `json:"default"`
	Min     *string      `json:"min"`
	Max     *string      `json:"max"`
	Options *[]string    `json:"options"`
	Help    *string      `json:"help"`
}

type RunItem struct {
	Command string `json:"command,omitempty" yaml:"command"`
	Mounts  []struct {
		Type   string `json:"type,omitempty" yaml:"type"`
		ID     string `json:"id,omitempty" yaml:"id"`
		Target string `json:"target,omitempty" yaml:"target"`
	} `json:"mounts,omitempty" yaml:"mounts,omitempty"`
}

type SchemaError struct {
	Field   string
	Message string
}

type TFCompatibility struct {
	TF           string
	TFCPUPackage string
	TFGPUPackage string
	CUDA         string
	CuDNN        string
	Pythons      []string
}

type TorchCompatibility struct {
	Torch         string
	Torchvision   string
	Torchaudio    string
	FindLinks     string
	ExtraIndexURL string
	CUDA          *string
	Pythons       []string
}

type ValidateOption func(*validateOptions)

type ValidationError struct {
	Field   string
	Value   string
	Message string
}

type ValidationResult struct {
	Errors   []error
	Warnings []DeprecationWarning
}

type WeightSource struct {
	Name   string `json:"name,omitempty" yaml:"name,omitempty"`
	Source string `json:"source" yaml:"source"`
	Target string `json:"target,omitempty" yaml:"target,omitempty"`
}