Skip to content

Commit

Permalink
Revert "new commits?"
Browse files Browse the repository at this point in the history
This reverts commit bfdc809.
  • Loading branch information
strakam committed Jan 20, 2024
1 parent bfdc809 commit 5445df1
Show file tree
Hide file tree
Showing 2 changed files with 13 additions and 17 deletions.
28 changes: 12 additions & 16 deletions aviaries/FollowerAviary.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,26 +6,24 @@
from gymnasium import spaces

from gym_pybullet_drones.control.DSLPIDControl import DSLPIDControl
from trajectories.square_linear_trajectory import SquareLinearTrajectory

class FollowerAviary(BaseRLAviary):
"""Single agent RL problem: hover at position."""

################################################################################

def __init__(
self,
drone_model: DroneModel=DroneModel.CF2X,
initial_xyzs=None,
initial_rpys=None,
physics: Physics=Physics.PYB,
pyb_freq: int = 240,
ctrl_freq: int = 30,
gui=False,
record=False,
obs: ObservationType=ObservationType.KIN,
act: ActionType=ActionType.RPM
):
def __init__(self,
drone_model: DroneModel=DroneModel.CF2X,
initial_xyzs=None,
initial_rpys=None,
physics: Physics=Physics.PYB,
pyb_freq: int = 240,
ctrl_freq: int = 30,
gui=False,
record=False,
obs: ObservationType=ObservationType.KIN,
act: ActionType=ActionType.RPM
):
"""Initialization of a single agent RL environment.
Using the generic single agent RL superclass.
Expand Down Expand Up @@ -59,8 +57,6 @@ def __init__(
self.NUM_DRONES = 1
self.WAYPOINT_BUFFER_SIZE = 3 # how many steps into future to interpolate
self.waypoint_buffer = np.zeros((self.WAYPOINT_BUFFER_SIZE,3)).reshape(1, -1)
self.trajectory = SquareLinearTrajectory()

super().__init__(
drone_model=drone_model,
num_drones=1,
Expand Down

0 comments on commit 5445df1

Please sign in to comment.