context

type API struct {
	*userconfig.API
	*ComputedResourceFields
	Path string `json:"path"`
}

type APIs map[string]*API

type Aggregate struct {
	*userconfig.Aggregate
	*ComputedResourceFields
	Type interface{} `json:"type"`
	Key  string      `json:"key"`
}

type Aggregates map[string]*Aggregate

type Aggregator struct {
	*userconfig.Aggregator
	*ResourceFields
	Namespace *string `json:"namespace"`
	ImplKey   string  `json:"impl_key"`
}

type Aggregators map[string]*Aggregator

type App struct {
	*userconfig.App
	ID string `json:"id"`
}

type Column interface {
	ComputedResource
	GetType() userconfig.ColumnType
	IsRaw() bool
	GetInputRawColumnNames() []string
}

type Columns map[string]Column

type ComputedResource interface {
	Resource
	GetWorkloadID() string
	SetWorkloadID(string)
}

type ComputedResourceFields struct {
	*ResourceFields
	WorkloadID string `json:"workload_id"`
}

type Constant struct {
	*userconfig.Constant
	*ResourceFields
	Key string `json:"key"`
}

type Constants map[string]*Constant

type Context struct {
	ID                 string             `json:"id"`
	Key                string             `json:"key"`
	CortexConfig       *CortexConfig      `json:"cortex_config"`
	DatasetVersion     string             `json:"dataset_version"`
	Root               string             `json:"root"`
	RawDataset         RawDataset         `json:"raw_dataset"`
	StatusPrefix       string             `json:"status_prefix"`
	App                *App               `json:"app"`
	Environment        *Environment       `json:"environment"`
	PythonPackages     PythonPackages     `json:"python_packages"`
	RawColumns         RawColumns         `json:"-"`
	Aggregates         Aggregates         `json:"aggregates"`
	TransformedColumns TransformedColumns `json:"transformed_columns"`
	Models             Models             `json:"models"`
	APIs               APIs               `json:"apis"`
	Constants          Constants          `json:"constants"`
	Aggregators        Aggregators        `json:"aggregators"`
	Transformers       Transformers       `json:"transformers"`
}

type CortexConfig struct {
	APIVersion string `json:"api_version"`
	Bucket     string `json:"bucket"`
	Region     string `json:"region"`
	LogGroup   string `json:"log_group"`
	ID         string `json:"id"`
}

type DataSplit struct {
	CSVData     *userconfig.CSVData     `json:"csv_data"`
	ParquetData *userconfig.ParquetData `json:"parquet_data"`
}

type Environment struct {
	*userconfig.Environment
	ID string `json:"id"`
}

type Model struct {
	*userconfig.Model
	*ComputedResourceFields
	Key     string           `json:"key"`
	ImplID  string           `json:"impl_id"`
	ImplKey string           `json:"impl_key"`
	Dataset *TrainingDataset `json:"dataset"`
}

type Models map[string]*Model

type PythonPackage struct {
	userconfig.ResourceConfigFields
	*ComputedResourceFields
	SrcKey     string `json:"src_key"`
	PackageKey string `json:"package_key"`
}

type PythonPackages map[string]*PythonPackage

type RawColumn interface {
	Column
	GetCompute() *userconfig.SparkCompute
	GetUserConfig() userconfig.Resource
}

type RawColumns map[string]RawColumn

type RawColumnsTypeSplit struct {
	RawIntColumns    map[string]*RawIntColumn    `json:"raw_int_columns"`
	RawStringColumns map[string]*RawStringColumn `json:"raw_string_columns"`
	RawFloatColumns  map[string]*RawFloatColumn  `json:"raw_float_columns"`
}

type RawDataset struct {
	Key         string `json:"key"`
	MetadataKey string `json:"metadata_key"`
}

type RawFloatColumn struct {
	*userconfig.RawFloatColumn
	*ComputedResourceFields
}

type RawIntColumn struct {
	*userconfig.RawIntColumn
	*ComputedResourceFields
}

type RawStringColumn struct {
	*userconfig.RawStringColumn
	*ComputedResourceFields
}

type Resource interface {
	userconfig.Resource
	GetID() string
	GetIDWithTags() string
	GetResourceFields() *ResourceFields
}

type ResourceFields struct {
	ID           string        `json:"id"`
	IDWithTags   string        `json:"id_with_tags"`
	ResourceType resource.Type `json:"resource_type"`
}

type Serial struct {
	Context
	RawColumnSplit *RawColumnsTypeSplit `json:"raw_columns"`
	DataSplit      *DataSplit           `json:"environment_data"`
}

type TrainingDataset struct {
	userconfig.ResourceConfigFields
	*ComputedResourceFields
	ModelName   string `json:"model_name"`
	TrainKey    string `json:"train_key"`
	EvalKey     string `json:"eval_key"`
	MetadataKey string `json:"metadata_key"`
}

type TrainingDatasets map[string]*TrainingDataset

type TransformedColumn struct {
	*userconfig.TransformedColumn
	*ComputedResourceFields
	Type userconfig.ColumnType `json:"type"`
}

type TransformedColumns map[string]*TransformedColumn

type Transformer struct {
	*userconfig.Transformer
	*ResourceFields
	Namespace *string `json:"namespace"`
	ImplKey   string  `json:"impl_key"`
}

type Transformers map[string]*Transformer

type ValueResource interface {
	Resource
	GetType() interface{}
}