pushworld/python3/scripts/demo_dm_env.py at main · google-deepmind/pushworld · GitHub

Repository navigation

History

51 lines (39 loc) · 1.37 KB

Raw

# Licensed under the Apache License, Version 2.0 (the "License");

# you may not use this file except in compliance with the License.

# You may obtain a copy of the License at

# http://www.apache.org/licenses/LICENSE-2.0

# Unless required by applicable law or agreed to in writing, software

# distributed under the License is distributed on an "AS IS" BASIS,

# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

# See the License for the specific language governing permissions and

# limitations under the License.

from absl import app

from absl import flags

import matplotlib

import matplotlib.pyplot as plt

import numpy as np

from pushworld.dm_env import PushWorldEnv

from pushworld.puzzle import NUM_ACTIONS

_PATH = flags.DEFINE_string('path', '', 'Puzzle file path.')

matplotlib.use('TkAgg')

def main(argv):

# Create PushWorldEnv

env = PushWorldEnv(_PATH.value)

# Reset the environment and show observation

timestep = env.reset()

plt.figure(figsize=(5, 5))

plt.imshow(timestep.observation)

plt.ion()

plt.show()

# Randomly take 10 actions and show observation

for _ in range(10):

timestep = env.step(np.random.randint(NUM_ACTIONS))

plt.imshow(timestep.observation)

plt.draw()

plt.pause(0.5)

if __name__ == '__main__':

app.run(main)

pushworld/python3/scripts/demo_dm_env.py at main · google-deepmind/pushworld · GitHub

Repository navigation

Expand file tree

Breadcrumbs

demo_dm_env.py

Latest commit

History

Breadcrumbs

demo_dm_env.py

File metadata and controls