def main(argv: typing.List[str]):
"""Main entry point."""
if len(argv) > 1:
raise app.UsageError("Unknown arguments: '{}'.".format(' '.join(
argv[1:])))
path = pathlib.Path(FLAGS.delayed_reward_experiment_path)
data = pbutil.FromFile(path, random_opt_pb2.DelayedRewardExperiment())
# graph = DelayedRewardExperimentToGraph(data)
# print(graph.ToDot())
dot = DelayedRewardExperimentToDot(data)
print(dot.source)
def ToProto(self) -> random_opt_pb2.DelayedRewardExperiment:
"""Return proto representation of environment."""
return random_opt_pb2.DelayedRewardExperiment(env=self.config,
episode=self.episodes)