arm/policy became pure virtual func

JoeLee2013 · Mar 14, 2014 · a6f05ba · a6f05ba
1 parent 3182608
commit a6f05ba
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 29 deletions.
diff --git a/arm/arm.hpp b/arm/arm.hpp
@@ -8,21 +8,9 @@ namespace bandit{
 class Arm{
 public:
   //base functions should not be called
-  virtual double pull(){
-    std::cout << "unexpected base Arm::pull function call" << std::endl;
-    abort();
-    return 0;
-	}
-  virtual double getExpectedReward(){
-    std::cout << "unexpected base Arm::getExpectedReward function call" << std::endl;
-    abort();
-    return 0;
-  }
-  virtual std::string toString(){
-    std::cout << "unexpected base Arm::toString function call" << std::endl;
-    abort();
-    return "";
-  }
+  virtual double pull() = 0;
+  virtual double getExpectedReward() = 0;
+  virtual std::string toString() = 0;
 };
 
 } //namespace
diff --git a/policy/policy.hpp b/policy/policy.hpp
@@ -9,20 +9,9 @@ namespace bandit{
 class Policy{
 public:
   //base functions should not be called
-  virtual int selectNextArm(){
-    std::cout << "unexpected base Policy::selectNextArm function call" << std::endl;
-    abort();
-    return -1;
-	}
-  virtual void updateState(int, double){
-    std::cout << "unexpected base Policy::updateState function call" << std::endl;
-    abort();
-  }
-  virtual std::string toString(){
-    std::cout << "unexpected base Policy::toString function call" << std::endl;
-    abort();
-    return "";
-  }
+  virtual int selectNextArm() = 0;
+  virtual void updateState(int, double) = 0;
+  virtual std::string toString() = 0;
 };
 
 } //namespace