environment.h

Go to the documentation of this file.

//=== MLModelRunner/ONNXModelRunner/environment.h - ONNX Environment C++ -===//
//
// Part of the MLCompilerBridge Project, under the Apache License v2.0 with LLVM
// Exceptions. See the LICENSE file for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===---------------------------------------------------------------------===//
//===----------------------------------------------------------------------===//
 
#ifndef ONNX_MODELRUNNER_ENVIRONMENT_H
#define ONNX_MODELRUNNER_ENVIRONMENT_H
 
#include "MLModelRunner/ONNXModelRunner/agent.h"
#include <map>
 
typedef signed Action;
 
namespace MLBridge {
class Environment {
  bool done = false;
  std::string nextAgent = "";
 
protected:
  std::map<std::string, Observation &> obsMap;
 
public:
  bool checkDone() { return done == true; };
 
  void setDone() { done = true; }
  void resetDone() { done = false; }
 
  std::string getNextAgent() { return nextAgent; };
 
  void setNextAgent(std::string name) { nextAgent = name; }
 
  virtual Observation &step(Action action) = 0;
 
  virtual Observation &reset() = 0;
};
} // namespace MLBridge
#endif // ONNX_MODELRUNNER_ENVIRONMENT_H