From 0c59eb75b2b1851046bcfa43de57bb5fd57dc48a Mon Sep 17 00:00:00 2001 From: Mark Stephenson Date: Sat, 1 Jun 2024 08:53:17 -0700 Subject: [PATCH] Fix docs build --- docs/source/_static/custom.css | 1 - src/bsk_rl/gym.py | 9 +++++++-- 2 files changed, 7 insertions(+), 3 deletions(-) diff --git a/docs/source/_static/custom.css b/docs/source/_static/custom.css index 81f08ea5..2c7d9875 100644 --- a/docs/source/_static/custom.css +++ b/docs/source/_static/custom.css @@ -29,7 +29,6 @@ a.el:visited { .wy-side-nav-search>div.version { color: #565A5C; - # CU dark gray } .wy-nav-content { diff --git a/src/bsk_rl/gym.py b/src/bsk_rl/gym.py index 24e4e5ad..024492cd 100644 --- a/src/bsk_rl/gym.py +++ b/src/bsk_rl/gym.py @@ -33,8 +33,8 @@ class GeneralSatelliteTasking(Env, Generic[SatObs, SatAct]): def __init__( self, satellites: Union[Satellite, list[Satellite]], - scenario: Scenario = Scenario(), - rewarder: GlobalReward = NoReward(), + scenario: Optional[Scenario] = None, + rewarder: Optional[GlobalReward] = None, world_type: Optional[type[WorldModel]] = None, world_args: Optional[dict[str, Any]] = None, communicator: Optional[CommunicationMethod] = None, @@ -91,6 +91,11 @@ def __init__( self.satellites = satellites self.simulator: Simulator + if scenario is None: + scenario = Scenario() + if rewarder is None: + rewarder = NoReward() + if world_type is None: world_type = self._minimum_world_model() self.world_type = world_type