Python API
Public packages and their exported classes, functions and constants.
mrkwatkins.oakemu_rl.gymnasiummrkwatkins.oakemu_rl.gymnasium.EpisodeOutcomemrkwatkins.oakemu_rl.gymnasium.GameConfigurationmrkwatkins.oakemu_rl.gymnasium.ModelConfigurationmrkwatkins.oakemu_rl.gymnasium.MultiEnvResultmrkwatkins.oakemu_rl.gymnasium.MultiEnvRunmrkwatkins.oakemu_rl.gymnasium.OakEmuEnvmrkwatkins.oakemu_rl.gymnasium.OakEmuEnvFactorymrkwatkins.oakemu_rl.gymnasium.OakEmuMultiEnvmrkwatkins.oakemu_rl.gymnasium.TrainingConfigurationmrkwatkins.oakemu_rl.gymnasium.Trajectorymrkwatkins.oakemu_rl.gymnasium.TrajectoryDatamrkwatkins.oakemu_rl.gymnasium.TrajectoryFilemrkwatkins.oakemu_rl.gymnasium.observation_layoutmrkwatkins.oakemu_rl.gymnasium.registrationmrkwatkins.oakemu_rl.gymnasium.unsafe_actionsmrkwatkins.oakemu_rl.gymnasium.unsafe_resultsmrkwatkins.oakemu_rl.frameworkmrkwatkins.oakemu_rl.framework.BackgroundCheckpointWritermrkwatkins.oakemu_rl.framework.BackgroundCheckpointWriter.Checkpointmrkwatkins.oakemu_rl.framework.BackgroundCheckpointWriter.CheckpointFilemrkwatkins.oakemu_rl.framework.BackgroundCheckpointWriter.CheckpointManagermrkwatkins.oakemu_rl.framework.BackgroundCheckpointWriter.MetricsTrackermrkwatkins.oakemu_rl.framework.CheckpointPolicymrkwatkins.oakemu_rl.framework.CheckpointPolicy.ExperimentRollbackControllermrkwatkins.oakemu_rl.framework.CheckpointPolicy.MetricsTrackermrkwatkins.oakemu_rl.framework.Configurationmrkwatkins.oakemu_rl.framework.EpisodeSchedulermrkwatkins.oakemu_rl.framework.EpisodeScheduler.MetricsTrackermrkwatkins.oakemu_rl.framework.EpisodeScheduler.OakEmuMultiEnvmrkwatkins.oakemu_rl.framework.Experimentmrkwatkins.oakemu_rl.framework.Experiment.AutoRollbackMonitormrkwatkins.oakemu_rl.framework.Experiment.Checkpointmrkwatkins.oakemu_rl.framework.Experiment.CheckpointFilemrkwatkins.oakemu_rl.framework.Experiment.CheckpointFileRegistrationmrkwatkins.oakemu_rl.framework.Experiment.CheckpointManagermrkwatkins.oakemu_rl.framework.Experiment.ExperimentRollbackControllermrkwatkins.oakemu_rl.framework.Experiment.GameConfigurationmrkwatkins.oakemu_rl.framework.Experiment.MetricsTrackermrkwatkins.oakemu_rl.framework.Experiment.MetricsWriterContextmrkwatkins.oakemu_rl.framework.Experiment.MetricsWriterFactorymrkwatkins.oakemu_rl.framework.Experiment.ModelConfigurationmrkwatkins.oakemu_rl.framework.Experiment.OakEmuMultiEnvmrkwatkins.oakemu_rl.framework.Experiment.Schedulemrkwatkins.oakemu_rl.framework.Experiment.TrainingConfigurationmrkwatkins.oakemu_rl.framework.ExperimentConfigurationmrkwatkins.oakemu_rl.framework.ExperimentConfiguration.CheckpointConfigurationmrkwatkins.oakemu_rl.framework.ExportModelmrkwatkins.oakemu_rl.framework.ObservationTransferWorkspacemrkwatkins.oakemu_rl.framework.OnnxExportermrkwatkins.oakemu_rl.framework.OnnxExporter.GameConfigurationmrkwatkins.oakemu_rl.framework.OnnxExporter.ModelConfigurationmrkwatkins.oakemu_rl.framework.OnnxExporter.TrainingConfigurationmrkwatkins.oakemu_rl.framework.RuntimeContextmrkwatkins.oakemu_rl.framework.RuntimeContext.GameConfigurationmrkwatkins.oakemu_rl.framework.RuntimeContext.MetricsTrackermrkwatkins.oakemu_rl.framework.RuntimeContext.ModelConfigurationmrkwatkins.oakemu_rl.framework.RuntimeContext.TrainingConfigurationmrkwatkins.oakemu_rl.framework.StateAdaptermrkwatkins.oakemu_rl.framework.StepPlanmrkwatkins.oakemu_rl.framework.ThreadSafeModelmrkwatkins.oakemu_rl.framework.TrainingExperimentConfigurationmrkwatkins.oakemu_rl.framework.TrainingExperimentConfiguration.MetricDropAutoRollbackConfigurationmrkwatkins.oakemu_rl.framework.TrainingExperimentConfiguration.Schedulemrkwatkins.oakemu_rl.framework.agentsmrkwatkins.oakemu_rl.framework.agents.ActionMaskermrkwatkins.oakemu_rl.framework.agents.ActionSelectormrkwatkins.oakemu_rl.framework.agents.Agentmrkwatkins.oakemu_rl.framework.agents.Agent.EpisodeOutcomemrkwatkins.oakemu_rl.framework.agents.Agent.MetricsTrackermrkwatkins.oakemu_rl.framework.agents.Agent.OakEmuMultiEnvmrkwatkins.oakemu_rl.framework.agents.EnvLifecyclemrkwatkins.oakemu_rl.framework.agents.EnvLifecycle.OakEmuMultiEnvmrkwatkins.oakemu_rl.framework.agents.EpisodeMetricsmrkwatkins.oakemu_rl.framework.agents.EpisodeMetrics.EpisodeOutcomemrkwatkins.oakemu_rl.framework.agents.EpisodeMetrics.MetricsTrackermrkwatkins.oakemu_rl.framework.agents.InferenceBatchermrkwatkins.oakemu_rl.framework.agents.ModelSwapCoordinatormrkwatkins.oakemu_rl.framework.agents.Testermrkwatkins.oakemu_rl.framework.agents.Tester.EpisodeOutcomemrkwatkins.oakemu_rl.framework.agents.Tester.MultiEnvResultmrkwatkins.oakemu_rl.framework.agents.Tester.OakEmuMultiEnvmrkwatkins.oakemu_rl.framework.auto_rollbackmrkwatkins.oakemu_rl.framework.auto_rollback.AutoRollbackConfigurationmrkwatkins.oakemu_rl.framework.auto_rollback.AutoRollbackMonitormrkwatkins.oakemu_rl.framework.auto_rollback.ExperimentRollbackControllermrkwatkins.oakemu_rl.framework.auto_rollback.MetricDropAutoRollbackConfigurationmrkwatkins.oakemu_rl.framework.checkpointsmrkwatkins.oakemu_rl.framework.checkpoints.Checkpointmrkwatkins.oakemu_rl.framework.checkpoints.CheckpointConfigurationmrkwatkins.oakemu_rl.framework.checkpoints.CheckpointFilemrkwatkins.oakemu_rl.framework.checkpoints.CheckpointFileRegistrationmrkwatkins.oakemu_rl.framework.checkpoints.CheckpointManagermrkwatkins.oakemu_rl.framework.checkpoints.ObjectCheckpointFilemrkwatkins.oakemu_rl.framework.checkpoints.StateDictSnapshotmrkwatkins.oakemu_rl.framework.checkpoints.TorchPayloadCheckpointFilemrkwatkins.oakemu_rl.framework.diagnosticsmrkwatkins.oakemu_rl.framework.diagnostics.ActivationStatisticsmrkwatkins.oakemu_rl.framework.diagnostics.ModelDiagnosticsmrkwatkins.oakemu_rl.framework.diagnostics.ObservationSamplermrkwatkins.oakemu_rl.framework.diagnostics.WeightStatisticsmrkwatkins.oakemu_rl.framework.diagnostics.rankmrkwatkins.oakemu_rl.framework.dqnmrkwatkins.oakemu_rl.framework.dqn.ActionPolicymrkwatkins.oakemu_rl.framework.dqn.ActionPolicy.OakEmuMultiEnvmrkwatkins.oakemu_rl.framework.dqn.CategoricalConfigurationmrkwatkins.oakemu_rl.framework.dqn.CategoricalProjectionmrkwatkins.oakemu_rl.framework.dqn.DQNAgentmrkwatkins.oakemu_rl.framework.dqn.DQNExperimentmrkwatkins.oakemu_rl.framework.dqn.DQNExperiment.Checkpointmrkwatkins.oakemu_rl.framework.dqn.DQNExperiment.CheckpointFileRegistrationmrkwatkins.oakemu_rl.framework.dqn.DQNExperiment.NoisyNetsConfigurationmrkwatkins.oakemu_rl.framework.dqn.DQNExperiment.ObjectCheckpointFilemrkwatkins.oakemu_rl.framework.dqn.DQNExperimentConfigurationmrkwatkins.oakemu_rl.framework.dqn.DQNExperimentConfiguration.EpsilonGreedyConfigurationmrkwatkins.oakemu_rl.framework.dqn.DQNExperimentConfiguration.EpsilonGreedyConfiguration.Schedulemrkwatkins.oakemu_rl.framework.dqn.DQNExperimentConfiguration.NoisyNetsConfigurationmrkwatkins.oakemu_rl.framework.dqn.DQNExportModelmrkwatkins.oakemu_rl.framework.dqn.DQNModelmrkwatkins.oakemu_rl.framework.dqn.DQNModelConfigurationmrkwatkins.oakemu_rl.framework.dqn.DQNModelConfiguration.NoisyNetsConfigurationmrkwatkins.oakemu_rl.framework.dqn.DQNStateAdaptermrkwatkins.oakemu_rl.framework.dqn.DQNStateCheckpointFilemrkwatkins.oakemu_rl.framework.dqn.EnvironmentStatemrkwatkins.oakemu_rl.framework.dqn.Learnermrkwatkins.oakemu_rl.framework.dqn.Learner.Schedulemrkwatkins.oakemu_rl.framework.dqn.LearnerBatchmrkwatkins.oakemu_rl.framework.dqn.LearnerThrottlermrkwatkins.oakemu_rl.framework.dqn.LearnerTransferWorkspacemrkwatkins.oakemu_rl.framework.dqn.NoisyLinearmrkwatkins.oakemu_rl.framework.dqn.ReplayWritermrkwatkins.oakemu_rl.framework.dqn.Trainermrkwatkins.oakemu_rl.framework.dqn.Trainer.ExplorationStrategyFactorymrkwatkins.oakemu_rl.framework.dqn.Trainer.ExplorationStrategyFactory.EpsilonGreedyConfigurationmrkwatkins.oakemu_rl.framework.dqn.Trainer.ExplorationStrategyFactory.EpsilonGreedyConfiguration.Schedulemrkwatkins.oakemu_rl.framework.dqn.Trainer.ExplorationStrategyFactory.NoisyNetsConfigurationmrkwatkins.oakemu_rl.framework.dqn.Trainer.OakEmuMultiEnvmrkwatkins.oakemu_rl.framework.dqn.TransitionAccumulatormrkwatkins.oakemu_rl.framework.dqn.TransitionPipelinemrkwatkins.oakemu_rl.framework.dqn.explorationmrkwatkins.oakemu_rl.framework.dqn.exploration.EpsilonGreedyStrategymrkwatkins.oakemu_rl.framework.dqn.exploration.EpsilonGreedyStrategy.Schedulemrkwatkins.oakemu_rl.framework.dqn.exploration.ExplorationStrategymrkwatkins.oakemu_rl.framework.dqn.exploration.ExplorationStrategyFactorymrkwatkins.oakemu_rl.framework.dqn.exploration.ExplorationStrategyFactory.EpsilonGreedyConfigurationmrkwatkins.oakemu_rl.framework.dqn.exploration.ExplorationStrategyFactory.EpsilonGreedyConfiguration.Schedulemrkwatkins.oakemu_rl.framework.dqn.exploration.ExplorationStrategyFactory.NoisyNetsConfigurationmrkwatkins.oakemu_rl.framework.dqn.exploration.NoisyNetsStrategymrkwatkins.oakemu_rl.framework.dqn.exploration.configurationmrkwatkins.oakemu_rl.framework.dqn.exploration.configuration.EpsilonGreedyConfigurationmrkwatkins.oakemu_rl.framework.dqn.exploration.configuration.EpsilonGreedyConfiguration.Schedulemrkwatkins.oakemu_rl.framework.dqn.exploration.configuration.ExplorationConfigurationmrkwatkins.oakemu_rl.framework.dqn.exploration.configuration.NoisyNetsConfigurationmrkwatkins.oakemu_rl.framework.dqn.learner_batchmrkwatkins.oakemu_rl.framework.dqn.learner_batch.Schedulemrkwatkins.oakemu_rl.framework.dqn.learner_lossmrkwatkins.oakemu_rl.framework.dqn.learner_metricsmrkwatkins.oakemu_rl.framework.dqn.learner_metrics.MetricsTrackermrkwatkins.oakemu_rl.framework.dqn.learner_targetmrkwatkins.oakemu_rl.framework.dqn.model_buildingmrkwatkins.oakemu_rl.framework.dqn.replay_buffermrkwatkins.oakemu_rl.framework.dqn.replay_buffer.NStepBuffermrkwatkins.oakemu_rl.framework.dqn.replay_buffer.PrioritizedExperienceReplayBuffermrkwatkins.oakemu_rl.framework.dqn.replay_buffer.ReplayBuffermrkwatkins.oakemu_rl.framework.dqn.replay_buffer.ReplayBufferCheckpointFilemrkwatkins.oakemu_rl.framework.dqn.replay_buffer.ReplayBufferCheckpointFile.CheckpointFilemrkwatkins.oakemu_rl.framework.dqn.replay_buffer.ReplayBufferFactorymrkwatkins.oakemu_rl.framework.dqn.replay_buffer.ReplayBufferFactory.Schedulemrkwatkins.oakemu_rl.framework.dqn.replay_buffer.StorageBackedReplayBuffermrkwatkins.oakemu_rl.framework.dqn.replay_buffer.SumTreemrkwatkins.oakemu_rl.framework.dqn.replay_buffer.TransitionStoragemrkwatkins.oakemu_rl.framework.dqn.replay_buffer.UniformReplayBuffermrkwatkins.oakemu_rl.framework.dqn.replay_buffer.configurationmrkwatkins.oakemu_rl.framework.dqn.replay_buffer.configuration.PrioritizedExperienceReplayBufferConfigurationmrkwatkins.oakemu_rl.framework.dqn.replay_buffer.configuration.PrioritizedExperienceReplayBufferConfiguration.Schedulemrkwatkins.oakemu_rl.framework.dqn.replay_buffer.configuration.ReplayBufferConfigurationmrkwatkins.oakemu_rl.framework.dqn.replay_buffer.configuration.UniformReplayBufferConfigurationmrkwatkins.oakemu_rl.framework.metricsmrkwatkins.oakemu_rl.framework.metrics.ConfigFlattenermrkwatkins.oakemu_rl.framework.metrics.ConsoleMetricsWritermrkwatkins.oakemu_rl.framework.metrics.ConsoleMetricsWriterConfigurationmrkwatkins.oakemu_rl.framework.metrics.MetricNamesmrkwatkins.oakemu_rl.framework.metrics.MetricsTrackermrkwatkins.oakemu_rl.framework.metrics.MetricsWritermrkwatkins.oakemu_rl.framework.metrics.MetricsWriterContextmrkwatkins.oakemu_rl.framework.metrics.MetricsWriterFactorymrkwatkins.oakemu_rl.framework.metrics.RateTrackermrkwatkins.oakemu_rl.framework.metrics.SystemMetricsReportermrkwatkins.oakemu_rl.framework.metrics.TensorboardMetricsWritermrkwatkins.oakemu_rl.framework.metrics.TensorboardMetricsWriterConfigurationmrkwatkins.oakemu_rl.framework.metrics.WindowedMetricmrkwatkins.oakemu_rl.framework.optimizationmrkwatkins.oakemu_rl.framework.optimization.Schedulemrkwatkins.oakemu_rl.framework.ppomrkwatkins.oakemu_rl.framework.ppo.BCSamplermrkwatkins.oakemu_rl.framework.ppo.GAEmrkwatkins.oakemu_rl.framework.ppo.PPOActivationMonitormrkwatkins.oakemu_rl.framework.ppo.PPOActivationMonitor.MetricsTrackermrkwatkins.oakemu_rl.framework.ppo.PPOActormrkwatkins.oakemu_rl.framework.ppo.PPOActor.EpisodeOutcomemrkwatkins.oakemu_rl.framework.ppo.PPOActor.OakEmuMultiEnvmrkwatkins.oakemu_rl.framework.ppo.PPOActorEnvironmentStatemrkwatkins.oakemu_rl.framework.ppo.PPOActorStepBuffersmrkwatkins.oakemu_rl.framework.ppo.PPOActorStepProcessormrkwatkins.oakemu_rl.framework.ppo.PPOAgentmrkwatkins.oakemu_rl.framework.ppo.PPOExperimentmrkwatkins.oakemu_rl.framework.ppo.PPOExperiment.Checkpointmrkwatkins.oakemu_rl.framework.ppo.PPOExperiment.CheckpointFilemrkwatkins.oakemu_rl.framework.ppo.PPOExperiment.CheckpointFileRegistrationmrkwatkins.oakemu_rl.framework.ppo.PPOExperimentConfigurationmrkwatkins.oakemu_rl.framework.ppo.PPOExperimentConfiguration.Schedulemrkwatkins.oakemu_rl.framework.ppo.PPOExperimentMilestoneLogmrkwatkins.oakemu_rl.framework.ppo.PPOExportModelmrkwatkins.oakemu_rl.framework.ppo.PPOLearnermrkwatkins.oakemu_rl.framework.ppo.PPOLearner.Schedulemrkwatkins.oakemu_rl.framework.ppo.PPOLearnerMetricsLoggermrkwatkins.oakemu_rl.framework.ppo.PPOLearnerMetricsLogger.MetricsTrackermrkwatkins.oakemu_rl.framework.ppo.PPOMinibatchUpdatermrkwatkins.oakemu_rl.framework.ppo.PPOModelmrkwatkins.oakemu_rl.framework.ppo.PPOModelConfigurationmrkwatkins.oakemu_rl.framework.ppo.PPOStateAdaptermrkwatkins.oakemu_rl.framework.ppo.PPOStateCheckpointFilemrkwatkins.oakemu_rl.framework.ppo.RolloutBuffermrkwatkins.oakemu_rl.framework.ppo.RolloutBufferMemoryDiagnosticsmrkwatkins.oakemu_rl.framework.ppo.RolloutFinalizermrkwatkins.oakemu_rl.framework.ppo.RolloutHandshakemrkwatkins.oakemu_rl.framework.restartsmrkwatkins.oakemu_rl.framework.restarts.CheckpointFilemrkwatkins.oakemu_rl.framework.restarts.Configurationmrkwatkins.oakemu_rl.framework.restarts.Configuration.Schedulemrkwatkins.oakemu_rl.framework.restarts.Poolmrkwatkins.oakemu_rl.framework.reward_shapingmrkwatkins.oakemu_rl.framework.reward_shaping.Pipelinemrkwatkins.oakemu_rl.framework.reward_shaping.Pipeline.Checkpointmrkwatkins.oakemu_rl.framework.reward_shaping.Pipeline.CheckpointFilemrkwatkins.oakemu_rl.framework.reward_shaping.Pipeline.ObjectCheckpointFilemrkwatkins.oakemu_rl.framework.reward_shaping.Shapermrkwatkins.oakemu_rl.framework.reward_shaping.ShaperMetricsmrkwatkins.oakemu_rl.framework.reward_shaping.typesmrkwatkins.oakemu_rl.framework.reward_shaping.types.MetricsTrackermrkwatkins.oakemu_rl.framework.schedulemrkwatkins.oakemu_rl.framework.schedule.ConstantSectionmrkwatkins.oakemu_rl.framework.schedule.InterpolatingSectionmrkwatkins.oakemu_rl.framework.schedule.Schedulemrkwatkins.oakemu_rl.framework.schedule.Sectionmrkwatkins.oakemu_rl.framework.step_result_helpersmrkwatkins.oakemu_rl.framework.trajectoriesmrkwatkins.oakemu_rl.framework.trajectories.BackwardCurriculummrkwatkins.oakemu_rl.framework.trajectories.BestRouteArchivemrkwatkins.oakemu_rl.framework.trajectories.CaptureMetricsmrkwatkins.oakemu_rl.framework.trajectories.CaptureMetrics.MetricsTrackermrkwatkins.oakemu_rl.framework.trajectories.CaptureProcessormrkwatkins.oakemu_rl.framework.trajectories.CaptureProcessor.MetricsTrackermrkwatkins.oakemu_rl.framework.trajectories.CellStatisticsmrkwatkins.oakemu_rl.framework.trajectories.Configurationmrkwatkins.oakemu_rl.framework.trajectories.Configuration.Schedulemrkwatkins.oakemu_rl.framework.trajectories.Datamrkwatkins.oakemu_rl.framework.trajectories.EpisodeCoordinatormrkwatkins.oakemu_rl.framework.trajectories.EpisodeCoordinator.MetricsTrackermrkwatkins.oakemu_rl.framework.trajectories.EpisodeTypeSchedulemrkwatkins.oakemu_rl.framework.trajectories.EpisodeTypeSchedule.Schedulemrkwatkins.oakemu_rl.framework.trajectories.FrameStacksmrkwatkins.oakemu_rl.framework.trajectories.MidpointStartPolicymrkwatkins.oakemu_rl.framework.trajectories.MidpointStartPolicy.Schedulemrkwatkins.oakemu_rl.framework.trajectories.ReplayObservationCachemrkwatkins.oakemu_rl.framework.trajectories.Repositorymrkwatkins.oakemu_rl.framework.trajectories.ResetParameterCurriculummrkwatkins.oakemu_rl.framework.trajectories.Retentionmrkwatkins.oakemu_rl.framework.trajectories.Seedingmrkwatkins.oakemu_rl.framework.trajectories.Selectionmrkwatkins.oakemu_rl.framework.trajectories.SourceSelectormrkwatkins.oakemu_rl.framework.trajectories.SourceSelector.MetricsTrackermrkwatkins.oakemu_rl.framework.trajectories.Storemrkwatkins.oakemu_rl.framework.trajectories.Utilitymrkwatkins.oakemu_rl.framework.trajectories.UtilityConfigurationmrkwatkins.oakemu_rl.framework.trajectories.checkpointsmrkwatkins.oakemu_rl.framework.trajectories.checkpoints.StoreCheckpointCellKeysmrkwatkins.oakemu_rl.framework.trajectories.checkpoints.StoreCheckpointCompatibilitymrkwatkins.oakemu_rl.framework.trajectories.checkpoints.StoreCheckpointFilemrkwatkins.oakemu_rl.framework.trajectories.checkpoints.StoreCheckpointSidecarArchivedRoutesmrkwatkins.oakemu_rl.framework.trajectories.checkpoints.StoreCheckpointSidecarBasemrkwatkins.oakemu_rl.framework.trajectories.checkpoints.StoreCheckpointSidecarBcFocusmrkwatkins.oakemu_rl.framework.trajectories.checkpoints.StoreCheckpointSidecarFrontiermrkwatkins.oakemu_rl.framework.trajectories.checkpoints.StoreCheckpointSidecarFrontierGridsmrkwatkins.oakemu_rl.framework.trajectories.checkpoints.StoreCheckpointSidecarKeysmrkwatkins.oakemu_rl.framework.trajectories.checkpoints.StoreCheckpointSidecarReplayBehaviourmrkwatkins.oakemu_rl.framework.trajectories.checkpoints.StoreCheckpointSnapshotmrkwatkins.oakemu_rl.imagemrkwatkins.oakemu_rl.image.imagemrkwatkins.oakemu_rl.image.typesmrkwatkins.oakemu_rl.onnxmrkwatkins.oakemu_rl.onnx.metadatamrkwatkins.oakemu_rl.onnx.metadata.GameConfigurationmrkwatkins.oakemu_rl.onnx.metadata.ModelConfigurationmrkwatkins.oakemu_rl.onnx.metadata.TrainingConfiguration