data-designer 0.3.5__tar.gz → 0.3.6__tar.gz

{data_designer-0.3.5 → data_designer-0.3.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: data-designer
-Version: 0.3.5
+Version: 0.3.6
 Summary: General framework for synthetic data generation
 License-Expression: Apache-2.0
 License-File: LICENSE

{data_designer-0.3.5 → data_designer-0.3.6}/src/data_designer/_version.py

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '0.3.5'
-__version_tuple__ = version_tuple = (0, 3, 5)
+__version__ = version = '0.3.6'
+__version_tuple__ = version_tuple = (0, 3, 6)
 
 __commit_id__ = commit_id = None

{data_designer-0.3.5 → data_designer-0.3.6}/src/data_designer/engine/models/litellm_overrides.py

@@ -7,9 +7,15 @@ import random
 import threading
 from typing import TYPE_CHECKING
 
+# Import specific litellm submodules needed for class inheritance
+# Note: Class inheritance requires base classes at definition time, so we import these directly.
+# Runtime litellm usage below still benefits from lazy loading via the litellm alias.
+import litellm.caching.in_memory_cache as _litellm_cache
+import litellm.router as _litellm_router
 from pydantic import BaseModel, Field
 from typing_extensions import override
 
+# Use lazy loading for runtime litellm usage (RetryPolicy, utils, etc.)
 from data_designer.lazy_heavy_imports import httpx, litellm
 from data_designer.logging import quiet_noisy_logger
 
@@ -48,7 +54,7 @@ class LiteLLMRouterDefaultKwargs(BaseModel):
     )
 
 
-class ThreadSafeCache(litellm.caching.in_memory_cache.InMemoryCache):
+class ThreadSafeCache(_litellm_cache.InMemoryCache):
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)
 
@@ -83,7 +89,7 @@ class ThreadSafeCache(litellm.caching.in_memory_cache.InMemoryCache):
             super().flush_cache()
 
 
-class CustomRouter(litellm.router.Router):
+class CustomRouter(_litellm_router.Router):
     def __init__(
         self,
         *args,

{data_designer-0.3.5 → data_designer-0.3.6}/tests/config/analysis/test_column_statistics.py

@@ -1,8 +1,10 @@
 # SPDX-FileCopyrightText: Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 # SPDX-License-Identifier: Apache-2.0
 
-import numpy as np
-import pandas as pd
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
 import pytest
 
 from data_designer.config.analysis.column_statistics import (
@@ -21,6 +23,11 @@ from data_designer.config.analysis.column_statistics import (
     SamplerType,
     ValidationColumnStatistics,
 )
+from data_designer.lazy_heavy_imports import np, pd
+
+if TYPE_CHECKING:
+    import numpy as np
+    import pandas as pd
 
 
 @pytest.fixture

{data_designer-0.3.5 → data_designer-0.3.6}/tests/config/test_config_builder.py

@@ -1,12 +1,14 @@
 # SPDX-FileCopyrightText: Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 # SPDX-License-Identifier: Apache-2.0
 
+from __future__ import annotations
+
 import json
 import tempfile
 from pathlib import Path
+from typing import TYPE_CHECKING
 from unittest.mock import patch
 
-import pandas as pd
 import pytest
 import yaml
 from pydantic import BaseModel, ValidationError
@@ -39,6 +41,10 @@ from data_designer.config.seed_source import DataFrameSeedSource, HuggingFaceSee
 from data_designer.config.utils.code_lang import CodeLang
 from data_designer.config.utils.info import ConfigBuilderInfo
 from data_designer.config.validator_params import CodeValidatorParams
+from data_designer.lazy_heavy_imports import pd
+
+if TYPE_CHECKING:
+    import pandas as pd
 
 
 class DummyStructuredModel(BaseModel):

{data_designer-0.3.5 → data_designer-0.3.6}/tests/config/test_seed_source.py

@@ -1,13 +1,19 @@
 # SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 # SPDX-License-Identifier: Apache-2.0
 
+from __future__ import annotations
+
 from pathlib import Path
+from typing import TYPE_CHECKING
 
-import pandas as pd
 import pytest
 
 from data_designer.config.errors import InvalidFilePathError
 from data_designer.config.seed_source import DataFrameSeedSource, LocalFileSeedSource
+from data_designer.lazy_heavy_imports import pd
+
+if TYPE_CHECKING:
+    import pandas as pd
 
 
 def create_partitions_in_path(temp_dir: Path, extension: str, num_files: int = 2) -> Path:

{data_designer-0.3.5 → data_designer-0.3.6}/tests/config/test_validator_params.py

@@ -1,7 +1,10 @@
 # SPDX-FileCopyrightText: Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 # SPDX-License-Identifier: Apache-2.0
 
-import pandas as pd
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
 import pytest
 from pydantic import ValidationError
 
@@ -11,6 +14,10 @@ from data_designer.config.validator_params import (
     LocalCallableValidatorParams,
     RemoteValidatorParams,
 )
+from data_designer.lazy_heavy_imports import pd
+
+if TYPE_CHECKING:
+    import pandas as pd
 
 
 def test_code_validator_params():

{data_designer-0.3.5 → data_designer-0.3.6}/tests/config/utils/test_io_helpers.py

@@ -1,18 +1,24 @@
 # SPDX-FileCopyrightText: Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 # SPDX-License-Identifier: Apache-2.0
 
+from __future__ import annotations
+
 import tempfile
 from datetime import date, datetime, timedelta
 from decimal import Decimal
 from pathlib import Path
+from typing import TYPE_CHECKING
 from unittest.mock import MagicMock, patch
 
-import numpy as np
-import pandas as pd
 import pytest
 import yaml
 
 from data_designer.config.utils.io_helpers import serialize_data, smart_load_dataframe, smart_load_yaml
+from data_designer.lazy_heavy_imports import np, pd
+
+if TYPE_CHECKING:
+    import numpy as np
+    import pandas as pd
 
 
 @patch("data_designer.config.utils.io_helpers.Path", autospec=True)

{data_designer-0.3.5 → data_designer-0.3.6}/tests/config/utils/test_visualization.py

@@ -1,7 +1,10 @@
 # SPDX-FileCopyrightText: Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 # SPDX-License-Identifier: Apache-2.0
 
-import pandas as pd
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
 import pytest
 
 from data_designer.config.config_builder import DataDesignerConfigBuilder
@@ -12,6 +15,10 @@ from data_designer.config.utils.visualization import (
     mask_api_key,
 )
 from data_designer.config.validator_params import CodeValidatorParams
+from data_designer.lazy_heavy_imports import pd
+
+if TYPE_CHECKING:
+    import pandas as pd
 
 
 @pytest.fixture

{data_designer-0.3.5 → data_designer-0.3.6}/tests/conftest.py

@@ -1,12 +1,14 @@
 # SPDX-FileCopyrightText: Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 # SPDX-License-Identifier: Apache-2.0
 
+from __future__ import annotations
+
 import os
 import tarfile
 import tempfile
 import textwrap
+from typing import TYPE_CHECKING
 
-import pandas as pd
 import pytest
 import yaml
 
@@ -18,6 +20,10 @@ from data_designer.config.data_designer_config import DataDesignerConfig
 from data_designer.config.models import ChatCompletionInferenceParams, ModelConfig, ModelProvider
 from data_designer.config.seed_source import HuggingFaceSeedSource
 from data_designer.engine.resources.seed_reader import SeedReader
+from data_designer.lazy_heavy_imports import pd
+
+if TYPE_CHECKING:
+    import pandas as pd
 
 
 @pytest.fixture

{data_designer-0.3.5 → data_designer-0.3.6}/tests/engine/analysis/column_profilers/test_base.py

@@ -1,13 +1,20 @@
 # SPDX-FileCopyrightText: Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 # SPDX-License-Identifier: Apache-2.0
 
-import pandas as pd
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
 import pytest
 from pydantic import ValidationError
 
 from data_designer.config.column_configs import SamplerColumnConfig
 from data_designer.config.sampler_params import SamplerType
 from data_designer.engine.analysis.column_profilers.base import ColumnConfigWithDataFrame
+from data_designer.lazy_heavy_imports import pd
+
+if TYPE_CHECKING:
+    import pandas as pd
 
 
 def test_column_config_with_dataframe_valid_column_config_with_dataframe():

{data_designer-0.3.5 → data_designer-0.3.6}/tests/engine/analysis/column_profilers/test_judge_score_profiler.py

@@ -1,10 +1,11 @@
 # SPDX-FileCopyrightText: Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 # SPDX-License-Identifier: Apache-2.0
 
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
 from unittest.mock import Mock, patch
 
-import pandas as pd
-import pyarrow as pa
 import pytest
 
 from data_designer.config.analysis.column_statistics import (
@@ -23,6 +24,11 @@ from data_designer.engine.analysis.column_profilers.judge_score_profiler import
     JudgeScoreSummary,
 )
 from data_designer.engine.analysis.utils.judge_score_processing import JudgeScoreDistributions, JudgeScoreSample
+from data_designer.lazy_heavy_imports import pa, pd
+
+if TYPE_CHECKING:
+    import pandas as pd
+    import pyarrow as pa
 
 
 @pytest.fixture

{data_designer-0.3.5 → data_designer-0.3.6}/tests/engine/analysis/conftest.py

@@ -1,12 +1,13 @@
 # SPDX-FileCopyrightText: Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 # SPDX-License-Identifier: Apache-2.0
 
+from __future__ import annotations
+
 import json
 from pathlib import Path
+from typing import TYPE_CHECKING
 from unittest.mock import Mock, patch
 
-import pandas as pd
-import pyarrow as pa
 from pytest import fixture
 
 from data_designer.config.analysis.column_statistics import (
@@ -26,6 +27,11 @@ from data_designer.engine.dataset_builders.artifact_storage import ArtifactStora
 from data_designer.engine.models.registry import ModelRegistry
 from data_designer.engine.registry.data_designer_registry import DataDesignerRegistry
 from data_designer.engine.resources.resource_provider import ResourceProvider
+from data_designer.lazy_heavy_imports import pa, pd
+
+if TYPE_CHECKING:
+    import pandas as pd
+    import pyarrow as pa
 
 
 @fixture

{data_designer-0.3.5 → data_designer-0.3.6}/tests/engine/analysis/test_column_statistics_calculator.py

@@ -1,7 +1,9 @@
 # SPDX-FileCopyrightText: Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 # SPDX-License-Identifier: Apache-2.0
 
-import pandas as pd
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
 
 from data_designer.config.analysis.column_statistics import ColumnDistributionType
 from data_designer.config.column_types import DataDesignerColumnType
@@ -9,6 +11,10 @@ from data_designer.config.sampler_params import SamplerType
 from data_designer.engine.analysis.column_profilers.base import ColumnConfigWithDataFrame
 from data_designer.engine.analysis.column_statistics import get_column_statistics_calculator
 from data_designer.engine.analysis.utils.column_statistics_calculations import ensure_hashable
+from data_designer.lazy_heavy_imports import pd
+
+if TYPE_CHECKING:
+    import pandas as pd
 
 
 def test_general_column_statistics(stub_df, column_configs):

{data_designer-0.3.5 → data_designer-0.3.6}/tests/engine/analysis/utils/test_column_statistics_calculations.py

@@ -1,11 +1,11 @@
 # SPDX-FileCopyrightText: Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 # SPDX-License-Identifier: Apache-2.0
 
+from __future__ import annotations
+
 from itertools import cycle
+from typing import TYPE_CHECKING
 
-import numpy as np
-import pandas as pd
-import pyarrow as pa
 import pytest
 
 from data_designer.config.analysis.column_statistics import (
@@ -27,6 +27,12 @@ from data_designer.engine.analysis.utils.column_statistics_calculations import (
     ensure_boolean,
     ensure_hashable,
 )
+from data_designer.lazy_heavy_imports import np, pa, pd
+
+if TYPE_CHECKING:
+    import numpy as np
+    import pandas as pd
+    import pyarrow as pa
 
 
 @pytest.fixture

{data_designer-0.3.5 → data_designer-0.3.6}/tests/engine/analysis/utils/test_judge_score_processing.py

@@ -1,7 +1,10 @@
 # SPDX-FileCopyrightText: Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 # SPDX-License-Identifier: Apache-2.0
 
-import pandas as pd
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
 import pytest
 
 from data_designer.config.analysis.column_statistics import (
@@ -18,6 +21,10 @@ from data_designer.engine.analysis.utils.judge_score_processing import (
     extract_judge_score_distributions,
     sample_scores_and_reasoning,
 )
+from data_designer.lazy_heavy_imports import pd
+
+if TYPE_CHECKING:
+    import pandas as pd
 
 
 def test_extract_judge_score_distributions_numerical_scores(stub_judge_column_config):

{data_designer-0.3.5 → data_designer-0.3.6}/tests/engine/column_generators/generators/test_column_generator_base.py

@@ -1,9 +1,10 @@
 # SPDX-FileCopyrightText: Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 # SPDX-License-Identifier: Apache-2.0
 
-from unittest.mock import Mock
+from __future__ import annotations
 
-import pandas as pd
+from typing import TYPE_CHECKING
+from unittest.mock import Mock
 
 from data_designer.config.column_configs import ExpressionColumnConfig
 from data_designer.engine.column_generators.generators.base import (
@@ -12,6 +13,10 @@ from data_designer.engine.column_generators.generators.base import (
     GenerationStrategy,
 )
 from data_designer.engine.resources.resource_provider import ResourceProvider
+from data_designer.lazy_heavy_imports import pd
+
+if TYPE_CHECKING:
+    import pandas as pd
 
 
 def _create_test_generator_class(strategy=GenerationStrategy.CELL_BY_CELL):

{data_designer-0.3.5 → data_designer-0.3.6}/tests/engine/column_generators/generators/test_expression.py

@@ -1,15 +1,21 @@
 # SPDX-FileCopyrightText: Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 # SPDX-License-Identifier: Apache-2.0
 
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
 from unittest.mock import Mock, patch
 
-import pandas as pd
 import pytest
 
 from data_designer.config.column_configs import ExpressionColumnConfig
 from data_designer.engine.column_generators.generators.expression import ExpressionColumnGenerator
 from data_designer.engine.column_generators.utils.errors import ExpressionTemplateRenderError
 from data_designer.engine.resources.resource_provider import ResourceProvider
+from data_designer.lazy_heavy_imports import pd
+
+if TYPE_CHECKING:
+    import pandas as pd
 
 
 def _create_test_config(name="test_column", expr="{{ col1 }}", dtype="str"):

{data_designer-0.3.5 → data_designer-0.3.6}/tests/engine/column_generators/generators/test_seed_dataset.py

@@ -1,12 +1,13 @@
 # SPDX-FileCopyrightText: Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 # SPDX-License-Identifier: Apache-2.0
 
+from __future__ import annotations
+
 import os
 import tempfile
+from typing import TYPE_CHECKING
 from unittest.mock import Mock, patch
 
-import duckdb
-import pandas as pd
 import pytest
 
 from data_designer.config.column_configs import SeedDatasetColumnConfig
@@ -20,6 +21,11 @@ from data_designer.engine.column_generators.generators.seed_dataset import (
 from data_designer.engine.column_generators.utils.errors import SeedDatasetError
 from data_designer.engine.dataset_builders.multi_column_configs import SeedDatasetMultiColumnConfig
 from data_designer.engine.resources.resource_provider import ResourceProvider
+from data_designer.lazy_heavy_imports import duckdb, pd
+
+if TYPE_CHECKING:
+    import duckdb
+    import pandas as pd
 
 
 @pytest.fixture

{data_designer-0.3.5 → data_designer-0.3.6}/tests/engine/column_generators/generators/test_validation.py

@@ -1,9 +1,11 @@
 # SPDX-FileCopyrightText: Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 # SPDX-License-Identifier: Apache-2.0
 
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
 from unittest.mock import Mock, patch
 
-import pandas as pd
 import pytest
 
 from data_designer.config.column_configs import ValidationColumnConfig
@@ -27,6 +29,10 @@ from data_designer.engine.validators import (
     ValidationResult,
 )
 from data_designer.engine.validators.base import ValidationOutput
+from data_designer.lazy_heavy_imports import pd
+
+if TYPE_CHECKING:
+    import pandas as pd
 
 
 @pytest.mark.parametrize(

{data_designer-0.3.5 → data_designer-0.3.6}/tests/engine/conftest.py

@@ -1,9 +1,11 @@
 # SPDX-FileCopyrightText: Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 # SPDX-License-Identifier: Apache-2.0
 
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
 from unittest.mock import Mock
 
-import pandas as pd
 import pytest
 
 from data_designer.config.run_config import RunConfig
@@ -12,6 +14,10 @@ from data_designer.engine.models.facade import ModelFacade
 from data_designer.engine.models.registry import ModelRegistry
 from data_designer.engine.resources.managed_storage import ManagedBlobStorage
 from data_designer.engine.resources.resource_provider import ResourceProvider
+from data_designer.lazy_heavy_imports import pd
+
+if TYPE_CHECKING:
+    import pandas as pd
 
 
 @pytest.fixture

{data_designer-0.3.5 → data_designer-0.3.6}/tests/engine/dataset_builders/test_artifact_storage.py

@@ -1,16 +1,22 @@
 # SPDX-FileCopyrightText: Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 # SPDX-License-Identifier: Apache-2.0
 
+from __future__ import annotations
+
 import json
 from datetime import datetime
+from typing import TYPE_CHECKING
 from unittest.mock import patch
 
-import pandas as pd
 import pytest
 from pyarrow import ArrowNotImplementedError
 
 from data_designer.engine.dataset_builders.artifact_storage import ArtifactStorage, BatchStage
 from data_designer.engine.dataset_builders.errors import ArtifactStorageError
+from data_designer.lazy_heavy_imports import pd
+
+if TYPE_CHECKING:
+    import pandas as pd
 
 
 @pytest.fixture

{data_designer-0.3.5 → data_designer-0.3.6}/tests/engine/dataset_builders/test_column_wise_builder.py

@@ -1,9 +1,11 @@
 # SPDX-FileCopyrightText: Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 # SPDX-License-Identifier: Apache-2.0
 
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
 from unittest.mock import Mock, patch
 
-import pandas as pd
 import pytest
 
 from data_designer.config.column_configs import LLMTextColumnConfig, SamplerColumnConfig
@@ -20,6 +22,10 @@ from data_designer.engine.dataset_builders.errors import DatasetGenerationError
 from data_designer.engine.models.telemetry import InferenceEvent, NemoSourceEnum, TaskStatusEnum
 from data_designer.engine.models.usage import ModelUsageStats, TokenUsageStats
 from data_designer.engine.registry.data_designer_registry import DataDesignerRegistry
+from data_designer.lazy_heavy_imports import pd
+
+if TYPE_CHECKING:
+    import pandas as pd
 
 
 @pytest.fixture

{data_designer-0.3.5 → data_designer-0.3.6}/tests/engine/dataset_builders/utils/test_dataset_batch_manager.py

@@ -1,15 +1,21 @@
 # SPDX-FileCopyrightText: Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 # SPDX-License-Identifier: Apache-2.0
 
+from __future__ import annotations
+
 import json
+from typing import TYPE_CHECKING
 from unittest.mock import Mock, patch
 
-import pandas as pd
 import pytest
 
 from data_designer.engine.dataset_builders.artifact_storage import BatchStage
 from data_designer.engine.dataset_builders.utils.dataset_batch_manager import DatasetBatchManager
 from data_designer.engine.dataset_builders.utils.errors import DatasetBatchManagementError
+from data_designer.lazy_heavy_imports import pd
+
+if TYPE_CHECKING:
+    import pandas as pd
 
 
 @pytest.fixture

{data_designer-0.3.5 → data_designer-0.3.6}/tests/engine/models/parsers/test_parser.py

@@ -1,6 +1,8 @@
 # SPDX-FileCopyrightText: Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 # SPDX-License-Identifier: Apache-2.0
 
+from __future__ import annotations
+
 from pydantic import BaseModel
 
 from data_designer.engine.models.parsers.parser import LLMResponseParser

{data_designer-0.3.5 → data_designer-0.3.6}/tests/engine/models/test_litellm_overrides.py

@@ -1,9 +1,11 @@
 # SPDX-FileCopyrightText: Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 # SPDX-License-Identifier: Apache-2.0
 
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
 from unittest.mock import patch
 
-import litellm
 import pytest
 
 from data_designer.engine.models.litellm_overrides import (
@@ -12,6 +14,10 @@ from data_designer.engine.models.litellm_overrides import (
     ThreadSafeCache,
     apply_litellm_patches,
 )
+from data_designer.lazy_heavy_imports import litellm
+
+if TYPE_CHECKING:
+    import litellm
 
 
 @pytest.fixture

{data_designer-0.3.5 → data_designer-0.3.6}/tests/engine/processing/processors/test_drop_columns.py

@@ -1,15 +1,21 @@
 # SPDX-FileCopyrightText: Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 # SPDX-License-Identifier: Apache-2.0
 
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
 from unittest.mock import Mock, patch
 
-import pandas as pd
 import pytest
 
 from data_designer.config.dataset_builders import BuildStage
 from data_designer.config.processors import DropColumnsProcessorConfig
 from data_designer.engine.dataset_builders.artifact_storage import BatchStage
 from data_designer.engine.processing.processors.drop_columns import DropColumnsProcessor
+from data_designer.lazy_heavy_imports import pd
+
+if TYPE_CHECKING:
+    import pandas as pd
 
 
 @pytest.fixture

{data_designer-0.3.5 → data_designer-0.3.6}/tests/engine/processing/processors/test_schema_transform.py

@@ -1,10 +1,12 @@
 # SPDX-FileCopyrightText: Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 # SPDX-License-Identifier: Apache-2.0
 
+from __future__ import annotations
+
 import json
+from typing import TYPE_CHECKING
 from unittest.mock import Mock
 
-import pandas as pd
 import pytest
 
 from data_designer.config.dataset_builders import BuildStage
@@ -12,6 +14,10 @@ from data_designer.config.processors import SchemaTransformProcessorConfig
 from data_designer.engine.dataset_builders.artifact_storage import BatchStage
 from data_designer.engine.processing.processors.schema_transform import SchemaTransformProcessor
 from data_designer.engine.resources.resource_provider import ResourceProvider
+from data_designer.lazy_heavy_imports import pd
+
+if TYPE_CHECKING:
+    import pandas as pd
 
 
 @pytest.fixture