env

type BoxSpace struct {
	// High values for this space.
	High *tensor.Dense

	// Low values for this space.
	Low *tensor.Dense

	// Shape of the space.
	Shape []int
}

type Env struct {
	*spherev1alpha.Environment

	// Client to connect to the Sphere server.
	Client spherev1alpha.EnvironmentAPIClient

	// VideoPaths of result videos downloadloaded from the server.
	VideoPaths []string

	// Normalizer normalizes observation data.
	Normalizer Normalizer

	// GoalNormalizer normalizes goal data.
	GoalNormalizer Normalizer
	// contains filtered or unexported fields
}

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

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

type InitialState struct {
	// Observation of the environment.
	Observation *tensor.Dense

	// Goal if present.
	Goal *tensor.Dense
}

type LocalServerConfig struct {
	// Docker image of environment.
	Image string

	// Version of the docker image.
	Version string

	// Port the environment is exposed on.
	Port string

	// Logger for the server.
	Logger *log.Logger
}

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

type Normalizer interface {
	// Init the normalizer.
	Init(env *Env) error

	// Norm normalizes the input data.
	Norm(input *tensor.Dense) (*tensor.Dense, error)
}

type Opt func(*Env)

type Outcome struct {
	// Observation of the current state.
	Observation *tensor.Dense

	// Action that was taken
	Action int

	// Reward from action.
	Reward float32

	// Whether the environment is done.
	Done bool
}

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

type Results struct {
	// Episodes is a map of episode id to result.
	Episodes map[int32]*spherev1alpha.EpisodeResult

	// Videos is a map of episode id to result.
	Videos map[int32]*spherev1alpha.Video

	// AverageReward is the average reward of the episodes.
	AverageReward float32
}

type Server struct {
	// Client to connect to the Sphere server.
	Client spherev1alpha.EnvironmentAPIClient
	// contains filtered or unexported fields
}

type ServerOpts func(*Server)