ujjwalpardeshi/pytorch-training-debugger

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