21424: Adjusts feature attributes for ordinal warning (#38)

howsoai · Aug 29, 2024 · 8ca74de · 8ca74de
1 parent 65a21a7
commit 8ca74de
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/howso_engine_rl_recipes/wafer_thin_mint/agent/basic_agent.py b/howso_engine_rl_recipes/wafer_thin_mint/agent/basic_agent.py
@@ -34,8 +34,8 @@ def setup(self) -> None:
         self.desired_score = 1
 
         self.features = {
-            'wafer_count': {'type': 'ordinal'},
-            'score': {'type': 'ordinal'},
+            'wafer_count': {'type': 'ordinal', 'data_type': 'number'},
+            'score': {'type': 'ordinal', 'data_type': 'number'},
             'action': {
                 'type': 'nominal',
                 'data_type': 'number',