Skip to content

Commit 767712e

Browse files
vertex-sdk-botcopybara-github
authored andcommitted
fix: Fix failing get_experiments_df test.
PiperOrigin-RevId: 617540149
1 parent 55f31b2 commit 767712e

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tests/system/aiplatform/test_experiments.py

+2
Original file line numberDiff line numberDiff line change
@@ -427,6 +427,8 @@ def test_get_experiments_df(self):
427427
true_df_dict_1["state"] = aiplatform.gapic.Execution.State.COMPLETE.name
428428
true_df_dict_1["run_type"] = aiplatform.metadata.constants.SYSTEM_EXPERIMENT_RUN
429429
true_df_dict_1[f"time_series_metric.{_TIME_SERIES_METRIC_KEY}"] = 4.0
430+
for i in range(_READ_TIME_SERIES_BATCH_SIZE + 1):
431+
true_df_dict_1[f"time_series_metric.{_TIME_SERIES_METRIC_KEY}-{i}"] = 1.0
430432

431433
true_df_dict_2 = {f"metric.{key}": value for key, value in _METRICS_2.items()}
432434
for key, value in _PARAMS_2.items():

0 commit comments

Comments
 (0)