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