(self, pipeline_state: pipeline.State)
| 62 | ) |
| 63 | |
| 64 | def _get_done(self, pipeline_state: pipeline.State) -> jnp.ndarray: |
| 65 | done = self._get_reward(pipeline_state) > 0.95 |
| 66 | return done.astype(jnp.float32) |
| 67 | |
| 68 | @property |
| 69 | def action_size(self): |
no test coverage detected