From de9774cedaa3f74311bbe43eda79c959a67e7f52 Mon Sep 17 00:00:00 2001
From: Louis Tiao <tiao@meta.com>
Date: Fri, 7 Feb 2025 15:59:44 -0800
Subject: [PATCH] Temporary Commit at 2/7/2025, 6:32:26 PM

Differential Revision: D69335324
---
 ax/preview/api/client.py            |  2 +-
 ax/preview/api/tests/test_client.py | 10 +++++-----
 2 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/ax/preview/api/client.py b/ax/preview/api/client.py
index f489a3664dc..16fed44b1ff 100644
--- a/ax/preview/api/client.py
+++ b/ax/preview/api/client.py
@@ -441,7 +441,7 @@ def attach_data(
         # If no progression is provided assume the data is not timeseries-like and
         # set step=NaN
         data_with_progression = [
-            ({"step": progression if progression is not None else np.nan}, raw_data)
+            ({"step": progression if progression is not None else -1.0}, raw_data)
         ]
 
         trial = assert_is_instance(self._experiment.trials[trial_index], Trial)
diff --git a/ax/preview/api/tests/test_client.py b/ax/preview/api/tests/test_client.py
index 0561bd8f5aa..13e416a882f 100644
--- a/ax/preview/api/tests/test_client.py
+++ b/ax/preview/api/tests/test_client.py
@@ -415,7 +415,7 @@ def test_attach_data(self) -> None:
                         "mean": {0: 1.0},
                         "sem": {0: np.nan},
                         "trial_index": {0: 0},
-                        "step": {0: np.nan},
+                        "step": {0: -1.0},
                     }
                 )
             ),
@@ -440,7 +440,7 @@ def test_attach_data(self) -> None:
                         "mean": {0: 1.0, 1: 2.0},
                         "sem": {0: np.nan, 1: np.nan},
                         "trial_index": {0: 0, 1: 0},
-                        "step": {0: np.nan, 1: 10.0},
+                        "step": {0: -1, 1: 10.0},
                     }
                 )
             ),
@@ -467,7 +467,7 @@ def test_attach_data(self) -> None:
                         "mean": {0: 2.0, 1: 1.0, 2: 2.0},
                         "sem": {0: np.nan, 1: np.nan, 2: np.nan},
                         "trial_index": {0: 0, 1: 0, 2: 0},
-                        "step": {0: 10.0, 1: np.nan, 2: np.nan},
+                        "step": {0: 10.0, 1: -1.0, 2: -1.0},
                     }
                 )
             ),
@@ -513,7 +513,7 @@ def test_complete_trial(self) -> None:
                         "mean": {0: 1.0, 1: 2.0},
                         "sem": {0: np.nan, 1: np.nan},
                         "trial_index": {0: 0, 1: 0},
-                        "step": {0: np.nan, 1: np.nan},
+                        "step": {0: -1.0, 1: -1.0},
                     }
                 )
             ),
@@ -568,7 +568,7 @@ def test_complete_trial(self) -> None:
                         "mean": {0: 1.0},
                         "sem": {0: np.nan},
                         "trial_index": {0: 2},
-                        "step": {0: np.nan},
+                        "step": {0: -1.0},
                     }
                 )
             ),