from gym.envs.registration import register
from d4rl.gym_mujoco import gym_envs

HOPPER_RANDOM_SCORE = -20.272305
HALFCHEETAH_RANDOM_SCORE = -280.178953
WALKER2D_RANDOM_SCORE = 1.629008
ANT_RANDOM_SCORE = -325.6
ANTCUSTOM_RANDOM_SCORE = 0.0
CHEETAHJUMP_RANDOM_SCORE = -1000.0

HOPPER_EXPERT_SCORE = 3234.3
HALFCHEETAH_EXPERT_SCORE = 12135.0
WALKER2D_EXPERT_SCORE = 4592.3
ANT_EXPERT_SCORE = 3879.7
ANTCUSTOM_EXPERT_SCORE = 1500.0
CHEETAHJUMP_EXPERT_SCORE = 2000.0

# Datasets and configs