Skip to content

GitLab

Unverified Commit 2fa3627a authored by Hongkuan Zhou's avatar Hongkuan Zhou Committed by GitHub
Browse files

test: fix missing args in replica calculation test (#5113) 


Signed-off-by: default avatarhongkuanz <hongkuanz@nvidia.com>
parent d7c27e68
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 1 deletion
tests/planner/test_replica_calculation.py
View file @ 2fa3627a
...@@ -40,6 +40,8 @@ from dynamo.planner.utils.planner_core import Metrics, Planner # noqa: E402 ...@@ -40,6 +40,8 @@ from dynamo.planner.utils.planner_core import Metrics, Planner # noqa: E402
@pytest.fixture @pytest.fixture
@pytest.mark.pre_merge
@pytest.mark.gpu_0
def planner(): def planner():
"""Set up test environment with mocked dependencies.""" """Set up test environment with mocked dependencies."""
# Create mock arguments # Create mock arguments
...@@ -54,7 +56,8 @@ def planner(): ...@@ -54,7 +56,8 @@ def planner():
args.backend = "vllm" args.backend = "vllm"
args.no_operation = True # Don't actually scale args.no_operation = True # Don't actually scale
args.no_correction = False # Allow correction factors args.no_correction = False # Allow correction factors
args.prometheus_port = 0 # 0 means disabled args.metric_pulling_prometheus_endpoint = "http://localhost:9090" # dummy endpoint
args.metric_reporting_prometheus_port = 0 # 0 means disabled
args.load_predictor = "constant" args.load_predictor = "constant"
args.load_prediction_window_size = 10 args.load_prediction_window_size = 10
args.profile_results_dir = os.path.join( args.profile_results_dir = os.path.join(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment