forked from pytorch/rl
-
Notifications
You must be signed in to change notification settings - Fork 0
/
__init__.py
110 lines (109 loc) · 2.39 KB
/
__init__.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
# Copyright (c) Meta Platforms, Inc. and affiliates.
#
# This source code is licensed under the MIT license found in the
# LICENSE file in the root directory of this source tree.
from .batched_envs import ParallelEnv, SerialEnv
from .common import EnvBase, EnvMetaData, make_tensordict
from .custom import PendulumEnv, TicTacToeEnv
from .env_creator import EnvCreator, get_env_metadata
from .gym_like import default_info_dict_reader, GymLikeEnv
from .libs import (
BraxEnv,
BraxWrapper,
DMControlEnv,
DMControlWrapper,
gym_backend,
GymEnv,
GymWrapper,
HabitatEnv,
IsaacGymEnv,
IsaacGymWrapper,
JumanjiEnv,
JumanjiWrapper,
MeltingpotEnv,
MeltingpotWrapper,
MOGymEnv,
MOGymWrapper,
MultiThreadedEnv,
MultiThreadedEnvWrapper,
OpenMLEnv,
PettingZooEnv,
PettingZooWrapper,
RoboHiveEnv,
set_gym_backend,
SMACv2Env,
SMACv2Wrapper,
VmasEnv,
VmasWrapper,
)
from .model_based import DreamerDecoder, DreamerEnv, ModelBasedEnvBase
from .transforms import (
ActionDiscretizer,
ActionMask,
AutoResetEnv,
AutoResetTransform,
BatchSizeTransform,
BinarizeReward,
BurnInTransform,
CatFrames,
CatTensors,
CenterCrop,
ClipTransform,
Compose,
DeviceCastTransform,
DiscreteActionProjection,
DoubleToFloat,
DTypeCastTransform,
EndOfLifeTransform,
ExcludeTransform,
FiniteTensorDictCheck,
FlattenObservation,
FrameSkipTransform,
GrayScale,
gSDENoise,
InitTracker,
KLRewardTransform,
MultiStepTransform,
NoopResetEnv,
ObservationNorm,
ObservationTransform,
PermuteTransform,
PinMemoryTransform,
R3MTransform,
RandomCropTensorDict,
RemoveEmptySpecs,
RenameTransform,
Resize,
Reward2GoTransform,
RewardClipping,
RewardScaling,
RewardSum,
SelectTransform,
SignTransform,
SqueezeTransform,
StepCounter,
TargetReturn,
TensorDictPrimer,
TimeMaxPool,
ToTensorImage,
Transform,
TransformedEnv,
UnsqueezeTransform,
VC1Transform,
VecGymEnvTransform,
VecNorm,
VIPRewardTransform,
VIPTransform,
)
from .utils import (
check_env_specs,
check_marl_grouping,
exploration_mode,
exploration_type,
ExplorationType,
make_composite_from_td,
MarlGroupMapType,
set_exploration_mode,
set_exploration_type,
step_mdp,
)