MushroomRL/mushroom-rl

from .angles import normalize_angle_positive, normalize_angle, shortest_angular_distance
from .angles import quat_to_euler, euler_to_quat, euler_to_mat, mat_to_euler
from .features import uniform_grid
from .frames import LazyFrames, preprocess_frame
from .numerical_gradient import numerical_diff_dist, numerical_diff_function, numerical_diff_policy
from .minibatches import minibatch_number, minibatch_generator
from .plot import plot_mean_conf, get_mean_and_confidence
from .record import VideoRecorder
from .torch import TorchUtils, CategoricalWrapper
from .viewer import Viewer, CV2Viewer, ImageViewer