fixes #31

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

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/explore/clr/vw_explore_clr_wrapper.h b/explore/clr/vw_explore_clr_wrapper.h
@@ -43,6 +43,13 @@ namespace MultiWorldTesting {
 		UInt64 JoinId;
 	};
 
+	public ref class ActionID
+	{
+	public:
+		static UInt32 Make_OneBased(UInt32 id) { return MWTAction::Make_OneBased(id); }
+		static UInt32 Make_ZeroBased(UInt32 id) { return MWTAction::Make_ZeroBased(id); }
+	};
+
 	public ref class MWTWrapper
 	{
 	private:

diff --git a/explore/tests/MWTExploreTests.cs b/explore/tests/MWTExploreTests.cs
@@ -60,15 +60,15 @@ private static UInt32 TestStatelessPolicyFunc(IntPtr applicationContext)
             GCHandle contextHandle = (GCHandle)applicationContext;
             CONTEXT context = (contextHandle.Target as CONTEXT);
 
-            return (uint)context.Features.Length % MWTExploreTests.NumActions + 1;
+            return ActionID.Make_OneBased((uint)context.Features.Length % MWTExploreTests.NumActions);
         }
 
         private static UInt32 TestStatefulPolicyFunc(IntPtr policyParams, IntPtr applicationContext)
         {
             GCHandle contextHandle = (GCHandle)applicationContext;
             CONTEXT context = (contextHandle.Target as CONTEXT);
 
-            return (uint)(policyParams + context.Features.Length) % MWTExploreTests.NumActions + 1;
+            return ActionID.Make_OneBased((uint)(policyParams + context.Features.Length) % MWTExploreTests.NumActions);
         }
 
         private static void TestStatefulScorerFunc(IntPtr policyParams, IntPtr applicationContext, float[] scores, uint size)