fix for #31

VowpalWabbit · Oct 6, 2014 · 6e49008 · 6e49008
1 parent e79bdd2
commit 6e49008
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/explore/static/Interaction.h b/explore/static/Interaction.h
@@ -85,7 +85,7 @@ class ActionSet
 	{
 		for (u32 i = 0; i < count; i++)
 		{
-			m_action_set.push_back(MWTAction(i + 1)); // 1-based Action id
+			m_action_set.push_back(MWTAction(MWTAction::Make_OneBased(i))); // 1-based Action id
 		}
 	}
 

diff --git a/explore/tests/MWTExploreTests.cpp b/explore/tests/MWTExploreTests.cpp
@@ -317,16 +317,16 @@ namespace vw_explore_tests
 	public:
 		static u32 Stateful_Default_Policy(int* policy_params, Context* applicationContext)
 		{
-			return *policy_params % m_num_actions + 1; // 1-based index
+			return MWTAction::Make_OneBased(*policy_params % m_num_actions);
 		}
 		static u32 Stateful_Default_Policy2(int* policy_params, Context* applicationContext)
 		{
-			return *policy_params % m_num_actions + 2; // 1-based index
+			return MWTAction::Make_OneBased(*policy_params % m_num_actions) + 1;
 		}
 
 		static u32 Stateless_Default_Policy(Context* applicationContext)
 		{
-			return 99 % m_num_actions + 1; // 1-based index
+			return MWTAction::Make_OneBased(99 % m_num_actions);
 		}
 
 		//TODO: For now assume the size of the score array is the number of action scores to