-
Notifications
You must be signed in to change notification settings - Fork 387
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
feature(wrh): add taxi env #799
Conversation
taxi env first commit
dizoo/taxi/envs/taxi_env.py
Outdated
def __init__(self, cfg: dict) -> None: | ||
|
||
#^ 该部分为初始化定义,需要有的 | ||
self._cfg = EasyDict( |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
use the implementation like default_config
in other previous environment in dizoo
dizoo/taxi/envs/taxi_env.py
Outdated
# Clean up temporary image files | ||
for temp_image_file in temp_image_files: | ||
os.remove(temp_image_file) | ||
print(f"GIF saved as {gif_path}") |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
use log instead of print.
|
||
@staticmethod | ||
def frames_to_gif(frames: List[imageio.core.util.Array], gif_path: str, duration: float = 0.1) -> None: | ||
""" |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Polish argument list as the format in https://github.com/opendilab/DI-engine/blob/main/ding/model/common/encoder.py.
Description
Related Issue
TODO
Check List