From d14130fba431227ceb7ffac6fe0ec845e866341d Mon Sep 17 00:00:00 2001 From: snow-fox Date: Mon, 11 Mar 2024 13:38:42 +0000 Subject: [PATCH] fix thing --- PyFlyt/gym_envs/rocket_envs/rocket_landing_env.py | 13 +++++++++---- 1 file changed, 9 insertions(+), 4 deletions(-) diff --git a/PyFlyt/gym_envs/rocket_envs/rocket_landing_env.py b/PyFlyt/gym_envs/rocket_envs/rocket_landing_env.py index c58fc928..992ebe73 100644 --- a/PyFlyt/gym_envs/rocket_envs/rocket_landing_env.py +++ b/PyFlyt/gym_envs/rocket_envs/rocket_landing_env.py @@ -2,6 +2,7 @@ from __future__ import annotations import os +from typing import Any import numpy as np import pybullet as p @@ -82,17 +83,21 @@ def __init__( """CONSTANTS""" self.sparse_reward = sparse_reward - def reset(self, seed=None, options=dict()): + def reset( + self, *, seed: None | int = None, options: None | dict[str, Any] = dict() + ): """Resets the environment. Args: seed: int options: None """ - options = dict(randomize_drop=True, accelerate_drop=True) - drone_options = dict(starting_fuel_ratio=0.01) + if options is None: + options = dict(randomize_drop=True, accelerate_drop=True) - super().begin_reset(seed, options, drone_options) + super().begin_reset( + seed=seed, options=options, drone_options=dict(starting_fuel_ratio=0.01) + ) # reset the tracked parameters self.landing_pad_contact = 0.0