unit test for rebalancing (#7271)
unit test for rebalancing (#7271)

* unit test for rebalancing

Signed-off-by: Prithvi Kannan <b6589def647aa7ef64f2412ba22eb67472d11fbb@databricks.com>

* check stderr

Signed-off-by: Prithvi Kannan <b6589def647aa7ef64f2412ba22eb67472d11fbb@databricks.com>

* whoops

Signed-off-by: Prithvi Kannan <b6589def647aa7ef64f2412ba22eb67472d11fbb@databricks.com>

Signed-off-by: Prithvi Kannan <b6589def647aa7ef64f2412ba22eb67472d11fbb@databricks.com>
mlflow/mlflow,mlflow/mlflow,mlflow/mlflow,mlflow/mlflow,mlflow/mlflow,mlflow/mlflow,mlflow/mlflow