File size: 629 Bytes
e2f8b29 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 | """Typed EnvClient for baseline scripts.
Extends GenericEnvClient since we can't easily subclass the
abstract EnvClient without implementing all transport methods.
Used by baseline_heuristic.py.
"""
from __future__ import annotations
from openenv.core.generic_client import GenericEnvClient
class MLTrainingEnvClient(GenericEnvClient):
"""Typed client for the PyTorch Training Debugger environment.
Wraps GenericEnvClient for convenient use in baselines.
Actions are sent as dicts matching MLTrainingAction schema.
Observations are received as dicts matching MLTrainingObservation schema.
"""
pass
|