Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/SurrealAI/surreal
/ functions
Functions
791 in github.com/SurrealAI/surreal
⨍
Functions
791
◇
Types & classes
148
↓ 40 callers
Method
reset
Resets the state of the environment and returns an initial observation. Returns: observation (object): the initial observation of the
surreal/env/base.py:96
↓ 40 callers
Method
step
Run one timestep of the environment's dynamics. When end of episode is reached, you are responsible for calling `reset()` to reset thi
surreal/env/base.py:78
↓ 40 callers
Method
time
(self)
surreal/utils/common.py:526
↓ 29 callers
Method
join
(self)
surreal/replay/base.py:62
↓ 24 callers
Method
extend
(self, default_config)
surreal/session/config.py:229
↓ 24 callers
Method
get_command
(self, role)
surreal/launch/command_generator.py:30
↓ 23 callers
Function
extend_config
default_config must specify all the expected keys. Use the following special values for required placeholders: * _req_: require a single
surreal/session/config.py:237
↓ 18 callers
Method
sum
Returns arr[start] + ... + arr[end]
surreal/replay/segment_tree.py:97
↓ 15 callers
Method
start
(self)
surreal/utils/common.py:534
↓ 14 callers
Function
pytest_print_raises
(exc)
test-old/utils.py:9
↓ 12 callers
Method
step
(self, action)
surreal/env/wrapper.py:52
↓ 10 callers
Method
numpy
(self, dtype=None)
surreal/env/atari_wrappers.py:210
↓ 9 callers
Method
add_value
(self, value)
surreal/utils/common.py:475
↓ 9 callers
Method
get
Returns a SharedMemoryObject whose .data attribute contains data
surreal/distributed/data_fetcher.py:63
↓ 9 callers
Method
load
Load data from a numpy_dictionary Args: numpy_dict: { <module_dict_key>: { <para
surreal/distributed/module_dict.py:47
↓ 9 callers
Method
main
Default Main loop Args: @env: the environment to run agent on
surreal/agent/base.py:224
↓ 9 callers
Method
render
Renders the environment.
surreal/env/base.py:105
↓ 8 callers
Method
__init__
(self, env)
surreal/env/wrapper.py:24
↓ 8 callers
Method
add_scalars
(self, new_data)
surreal/utils/common.py:637
↓ 8 callers
Method
build
Build all images and push them
surreal/launch/build_images.py:70
↓ 8 callers
Method
parse_args
(self)
surreal/subproc/surreal_subproc.py:59
↓ 8 callers
Method
track_increment
(self)
surreal/utils/common.py:596
↓ 8 callers
Method
update
(self, other)
surreal/session/config.py:182
↓ 7 callers
Method
__init__
This object ensures that common frames between the observations are only stored once. It exists purely to optimize memory usage which can be h
surreal/env/atari_wrappers_deprecated.py:157
↓ 7 callers
Method
close
Override _close in your subclass to perform any necessary cleanup. Environments will automatically close() themselves when garbage co
surreal/env/base.py:110
↓ 7 callers
Method
observation_spec
(self)
surreal/env/wrapper.py:86
↓ 6 callers
Method
ckpt_name
(self, suffix)
surreal/utils/checkpoint.py:221
↓ 6 callers
Method
forward
(self, x)
surreal/model/q_net.py:112
↓ 6 callers
Method
likelihood
Method computes likelihood of action (a) given probability (prob)
surreal/model/ppo_net.py:42
↓ 6 callers
Method
metadata_path
(self)
surreal/utils/checkpoint.py:218
↓ 5 callers
Method
_get_path
(self, fname)
surreal/utils/checkpoint.py:212
↓ 5 callers
Function
_trace_key
(dict_trace, key)
surreal/session/config.py:11
↓ 5 callers
Method
action_spec
(self)
surreal/env/wrapper.py:83
↓ 5 callers
Method
forward_actor
forward pass actor to generate policy with option to use z-filter Args: obs: dictionary of batched observatio
surreal/model/ppo_net.py:253
↓ 5 callers
Method
forward_critic
forward pass critic to generate policy with option to use z-filter Note: assumes input has either shape length 2 or 4 without
surreal/model/ppo_net.py:284
↓ 5 callers
Function
get_tensorplex_client
(client_id, session_config)
surreal/session/tracker.py:193
↓ 5 callers
Method
send
(self, data)
surreal/env/exp_sender_wrapper.py:56
↓ 4 callers
Method
__init__
(self, D_in, D_act, hidden_sizes=[300, 200], use_layernorm=True)
surreal/model/model_builders/builders.py:36
↓ 4 callers
Method
__init__
Stack k last frames. Returns lazy array, which is much more memory efficient. See Also -------- baselines.common.ata
surreal/env/atari_wrappers.py:149
↓ 4 callers
Method
__init__
Default sender configs are in BASE_SESSION_CONFIG['sender'] They contain communication level information Algorithm s
surreal/env/exp_sender_wrapper.py:11
↓ 4 callers
Method
_gather_low_dim_input
Concatenate (along 2nd dimension) all the low-dimensional (propioceptive) features from input observation tuple A
surreal/model/ppo_net.py:168
↓ 4 callers
Function
_is_req
(value)
surreal/session/config.py:56
↓ 4 callers
Method
_load_metadata
(self)
surreal/utils/checkpoint.py:97
↓ 4 callers
Method
copy
(self)
surreal/session/config.py:233
↓ 4 callers
Method
cur_value
Returns current moving average, 0 if no records
surreal/utils/common.py:484
↓ 4 callers
Method
get_value
(self)
test-old/test_redis.py:15
↓ 4 callers
Method
insert
Add a new experience to the replay. Includes passive evict logic if memory capacity is exceeded. Args: exp_dict:
surreal/replay/base.py:70
↓ 4 callers
Function
integration_test
(temp_path, config_path, launcher, env='gym:Hal
surreal/test_helpers/integration_test.py:42
↓ 4 callers
Method
launch
Launches the specific component Args: component_name(str): the process to launch
surreal/launch/launcher.py:58
↓ 4 callers
Function
make_env
Makes an environment and populates related fields in env_config return env, env_config Args: mode(str): allow differnt kwargs to
surreal/env/make_env.py:41
↓ 4 callers
Function
make_env_config
Issue: Surreal-tmux gives segfault on running robosuite environments. Cause: On launching a surreal process, an instance of RL environment is
surreal/env/make_env.py:16
↓ 4 callers
Method
run_component
(self, component_name)
surreal/launch/launcher.py:175
↓ 4 callers
Function
setup_network
Sets up the communication between surreal components using symphony Args: agents, evals (list): list of symphony
surreal/launch/setup_network.py:4
↓ 4 callers
Method
to_dict
Recursively convert back to dictionary
surreal/session/config.py:201
↓ 3 callers
Method
__init__
The following options are pre-configured --verbosity, or -vvv (number of v's indicate the level of verbosity) --debug: turn o
surreal/utils/common.py:368
↓ 3 callers
Method
__init__
(self, env)
surreal/env/monitor.py:20
↓ 3 callers
Method
_add_modality
(self, obs)
surreal/env/dm_wrapper.py:47
↓ 3 callers
Method
_add_modality
(self, obs)
surreal/env/wrapper.py:172
↓ 3 callers
Method
_add_modality
(self, obs, verbose=False)
surreal/env/wrapper.py:225
↓ 3 callers
Method
_check_nfs_retrieve_settings
(self)
surreal/kube/surreal_kube.py:546
↓ 3 callers
Method
_filtered_obs
(self, obs, verbose=False)
surreal/env/wrapper.py:483
↓ 3 callers
Method
_gcloud_download
(self, remote_path, local_path)
surreal/kube/surreal_kube.py:533
↓ 3 callers
Method
_scale_image
Given uint8 input from the environment, scale to float32 and divide by 255 to scale inputs between 0.0 and 1.0 Ar
surreal/model/ppo_net.py:368
↓ 3 callers
Method
_step
(self, action)
surreal/env/monitor.py:35
↓ 3 callers
Method
add_value
(self, x)
test-old/test_redis.py:11
↓ 3 callers
Function
assert_type
(x, expected_type, message='')
surreal/utils/common.py:37
↓ 3 callers
Function
build_from_config
(config)
surreal/env/mujocomanip/object_builder.py:3
↓ 3 callers
Method
ckpt_path
(self, suffix)
surreal/utils/checkpoint.py:228
↓ 3 callers
Method
dump_file
(self, file_path)
surreal/session/config.py:216
↓ 3 callers
Method
fetch_parameter
Extends base class fetch_parameters to add some logging
surreal/agent/base.py:355
↓ 3 callers
Method
forward_perception
(self, obs)
surreal/model/ddpg_net.py:69
↓ 3 callers
Method
get_actor_params
Method that returns generator that contains all the parameters from actor, cnn_stem (optional), and rnn_stem (optional)
surreal/model/ppo_net.py:202
↓ 3 callers
Function
get_loggerplex_client
Args: name: file name for the remote log file session_config: see session_config.loggerplex
surreal/session/tracker.py:173
↓ 3 callers
Method
join
Wait for all parameter server workers to exit (Currently this means they crashed) Note that proxy is a daemon
surreal/distributed/parameter_server.py:101
↓ 3 callers
Method
join
(self)
surreal/replay/sharded_replay.py:105
↓ 3 callers
Method
kl
Method computes KL Divergence of between two probability distributions Note: this is D_KL(prob0 || prob1), not D_KL(prob1 ||
surreal/model/ppo_net.py:48
↓ 3 callers
Method
main_loop
One loop of agent, runs one episode of the environment
surreal/agent/base.py:244
↓ 3 callers
Method
main_setup
Setup before constant looping
surreal/agent/base.py:234
↓ 3 callers
Method
min
Returns min(arr[start], ..., arr[end])
surreal/replay/segment_tree.py:135
↓ 3 callers
Method
process
(self, frame)
surreal/env/atari_wrappers_deprecated.py:132
↓ 3 callers
Method
step_per_sec
Speedometer, step per second Args: average_episodes: average over the past N episodes
surreal/env/monitor.py:59
↓ 3 callers
Method
train
(self)
test-old/test_checkpoint.py:72
↓ 2 callers
Method
__init__
(self, config)
surreal/env/mujocomanip/mujocomanip_envs.py:6
↓ 2 callers
Method
__init__
Build a Segment Tree data structure. https://en.wikipedia.org/wiki/Segment_tree Can be used as regular array, but with two imp
surreal/replay/segment_tree.py:7
↓ 2 callers
Method
_batch_obs
Helper function that batches a list of observation (nested dictionary) into one whole dictionary of batched observation
surreal/learner/aggregator.py:186
↓ 2 callers
Method
_change_folder
Temporary change save folder, useful for loading from other folders
surreal/utils/checkpoint.py:127
↓ 2 callers
Method
_check_version
(self)
surreal/utils/checkpoint.py:81
↓ 2 callers
Function
_fill_default_config
(config, default_config, dict_trace)
surreal/session/config.py:100
↓ 2 callers
Method
_flatten_obs
(self, obs)
surreal/env/wrapper.py:299
↓ 2 callers
Method
_get_ob
(self)
surreal/env/atari_wrappers_deprecated.py:203
↓ 2 callers
Method
_get_ob
(self)
surreal/env/atari_wrappers.py:182
↓ 2 callers
Function
_get_qualified_type_name
(type_)
surreal/utils/common.py:27
↓ 2 callers
Method
_get_tensorplex
Get the periodic tensorplex object Args: @name: The name of the collection of metrics
surreal/agent/base.py:108
↓ 2 callers
Method
_grayscale
(self, obs)
surreal/env/wrapper.py:370
↓ 2 callers
Function
_merge_setting_dictionaries
(customize, base)
surreal/kube/surreal_kube.py:20
↓ 2 callers
Method
_observation
(self, observation)
surreal/env/wrapper.py:109
↓ 2 callers
Function
_raise_req_error
(key, value, dict_trace, prefix_msg='')
surreal/session/config.py:70
↓ 2 callers
Function
_req_type_check
(value)
surreal/session/config.py:24
↓ 2 callers
Method
_reset
(self)
surreal/env/mujocomanip/mujocomanip_envs.py:10
↓ 2 callers
Method
_restore
helper to be used in self.restore() and self.restore_full_name() Returns: - ckpt file name if successfully restored
surreal/utils/checkpoint.py:102
next →
1–100 of 791, ranked by callers