Unverified Commit ab5a31b5 authored by Alec's avatar Alec Committed by GitHub
Browse files

test(planner): isolate planner-family suites [DYN-2534] (#7723)

parent cc22114d
...@@ -23,7 +23,12 @@ from dynamo.planner.core.state import PlannerSharedState ...@@ -23,7 +23,12 @@ from dynamo.planner.core.state import PlannerSharedState
from dynamo.planner.monitoring.traffic_metrics import Metrics from dynamo.planner.monitoring.traffic_metrics import Metrics
from dynamo.planner.monitoring.worker_info import WorkerInfo from dynamo.planner.monitoring.worker_info import WorkerInfo
pytestmark = [pytest.mark.pre_merge, pytest.mark.gpu_0] pytestmark = [
pytest.mark.pre_merge,
pytest.mark.gpu_0,
pytest.mark.unit,
pytest.mark.planner,
]
class PlannerHarness: class PlannerHarness:
...@@ -160,7 +165,10 @@ def planner(): ...@@ -160,7 +165,10 @@ def planner():
load_predictor="constant", load_predictor="constant",
profile_results_dir=os.path.join( profile_results_dir=os.path.join(
os.path.dirname(__file__), os.path.dirname(__file__),
"profiling_results/H200_TP1P_TP1D", "..",
"data",
"profiling_results",
"H200_TP1P_TP1D",
), ),
environment="kubernetes", environment="kubernetes",
namespace="test-namespace", namespace="test-namespace",
......
...@@ -22,7 +22,6 @@ pytestmark = [ ...@@ -22,7 +22,6 @@ pytestmark = [
pytest.mark.pre_merge, pytest.mark.pre_merge,
pytest.mark.unit, pytest.mark.unit,
pytest.mark.planner, pytest.mark.planner,
pytest.mark.vllm,
] ]
...@@ -53,6 +52,7 @@ def _build_config(): ...@@ -53,6 +52,7 @@ def _build_config():
profile_results_dir=os.path.join( profile_results_dir=os.path.join(
os.path.dirname(__file__), os.path.dirname(__file__),
"..", "..",
"data",
"profiling_results", "profiling_results",
"H200_TP1P_TP1D", "H200_TP1P_TP1D",
), ),
...@@ -406,6 +406,7 @@ class TestDryrunGpuDefaults: ...@@ -406,6 +406,7 @@ class TestDryrunGpuDefaults:
profile_results_dir=os.path.join( profile_results_dir=os.path.join(
os.path.dirname(__file__), os.path.dirname(__file__),
"..", "..",
"data",
"profiling_results", "profiling_results",
"H200_TP1P_TP1D", "H200_TP1P_TP1D",
), ),
......
...@@ -12,7 +12,6 @@ All tests are no-GPU (gpu_0) and pre_merge. ...@@ -12,7 +12,6 @@ All tests are no-GPU (gpu_0) and pre_merge.
import asyncio import asyncio
import os import os
import sys
from pathlib import Path from pathlib import Path
from unittest.mock import AsyncMock, MagicMock, patch from unittest.mock import AsyncMock, MagicMock, patch
...@@ -20,8 +19,12 @@ import numpy as np ...@@ -20,8 +19,12 @@ import numpy as np
import pytest import pytest
import yaml import yaml
project_root = Path(__file__).parent.parent.parent pytestmark = [
sys.path.insert(0, str(project_root)) pytest.mark.pre_merge,
pytest.mark.gpu_0,
pytest.mark.integration,
pytest.mark.planner,
]
try: try:
from dynamo.profiler.profile_sla import run_profile from dynamo.profiler.profile_sla import run_profile
...@@ -56,7 +59,7 @@ def _make_ops(tmp_path, **overrides) -> ProfilerOperationalConfig: ...@@ -56,7 +59,7 @@ def _make_ops(tmp_path, **overrides) -> ProfilerOperationalConfig:
return ProfilerOperationalConfig(**defaults) return ProfilerOperationalConfig(**defaults)
CONFIGS_DIR = Path(__file__).parent / "configs" CONFIGS_DIR = Path(__file__).parent.parent / "data" / "configs"
class TestRapidSupported: class TestRapidSupported:
......
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