data-designer 0.3.4__py3-none-any.whl → 0.3.6__py3-none-any.whl

data_designer/engine/sampling_gen/constraints.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 abc import ABC, abstractmethod
+from typing import TYPE_CHECKING
 
-import numpy as np
-import pandas as pd
 from numpy.typing import NDArray
 
 from data_designer.config.base import ConfigBase
@@ -15,6 +16,11 @@ from data_designer.config.sampler_constraints import (
     InequalityOperator,
     ScalarInequalityConstraint,
 )
+from data_designer.lazy_heavy_imports import np, pd
+
+if TYPE_CHECKING:
+    import numpy as np
+    import pandas as pd
 
 
 class ConstraintChecker(ConfigBase, ABC):

data_designer/engine/sampling_gen/data_sources/base.py

@@ -1,24 +1,27 @@
 # SPDX-FileCopyrightText: Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 # SPDX-License-Identifier: Apache-2.0
 
+from __future__ import annotations
+
 from abc import ABC, abstractmethod
-from typing import Any, Generic, TypeVar
+from typing import TYPE_CHECKING, Any, Generic, TypeVar
 
-import numpy as np
-import pandas as pd
 from numpy.typing import NDArray
-from scipy import stats
 
 from data_designer.config.sampler_params import SamplerParamsT
 from data_designer.engine.sampling_gen.utils import check_random_state
+from data_designer.lazy_heavy_imports import np, pd, scipy
+
+if TYPE_CHECKING:
+    import numpy as np
+    import pandas as pd
+    import scipy
 
 NumpyArray1dT = NDArray[Any]
 RadomStateT = int | np.random.RandomState
 
-
 GenericParamsT = TypeVar("GenericParamsT", bound=SamplerParamsT)
 
-
 ###########################################################
 # Processing Mixins
 # -----------------
@@ -208,7 +211,7 @@ class Sampler(DataSource[GenericParamsT], ABC):
 class ScipyStatsSampler(Sampler[GenericParamsT], ABC):
     @property
     @abstractmethod
-    def distribution(self) -> stats.rv_continuous | stats.rv_discrete: ...
+    def distribution(self) -> scipy.stats.rv_continuous | scipy.stats.rv_discrete: ...
 
     def sample(self, num_samples: int) -> NumpyArray1dT:
         return self.distribution.rvs(size=num_samples, random_state=self.rng)

data_designer/engine/sampling_gen/data_sources/errors.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 data_designer.engine.sampling_gen.errors import SamplingGenError
 
 

data_designer/engine/sampling_gen/data_sources/sources.py

@@ -1,11 +1,10 @@
 # SPDX-FileCopyrightText: Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 # SPDX-License-Identifier: Apache-2.0
 
-import uuid
+from __future__ import annotations
 
-import numpy as np
-import pandas as pd
-from scipy import stats
+import uuid
+from typing import TYPE_CHECKING
 
 from data_designer.config.sampler_params import (
     BernoulliMixtureSamplerParams,
@@ -40,6 +39,12 @@ from data_designer.engine.sampling_gen.data_sources.errors import (
 )
 from data_designer.engine.sampling_gen.entities.dataset_based_person_fields import PERSONA_FIELDS, PII_FIELDS
 from data_designer.engine.sampling_gen.people_gen import PeopleGen
+from data_designer.lazy_heavy_imports import np, pd, scipy
+
+if TYPE_CHECKING:
+    import numpy as np
+    import pandas as pd
+    import scipy
 
 ONE_BILLION = 10**9
 
@@ -264,8 +269,8 @@ class ScipySampler(TypeConversionMixin, ScipyStatsSampler[ScipySamplerParams]):
     """Escape hatch sampler to give users access to any scipy.stats distribution."""
 
     @property
-    def distribution(self) -> stats.rv_continuous | stats.rv_discrete:
-        return getattr(stats, self.params.dist_name)(**self.params.dist_params)
+    def distribution(self) -> scipy.stats.rv_continuous | scipy.stats.rv_discrete:
+        return getattr(scipy.stats, self.params.dist_name)(**self.params.dist_params)
 
     def _validate(self) -> None:
         _validate_scipy_distribution(self.params.dist_name, self.params.dist_params)
@@ -274,16 +279,16 @@ class ScipySampler(TypeConversionMixin, ScipyStatsSampler[ScipySamplerParams]):
 @SamplerRegistry.register(SamplerType.BERNOULLI)
 class BernoulliSampler(TypeConversionMixin, ScipyStatsSampler[BernoulliSamplerParams]):
     @property
-    def distribution(self) -> stats.rv_discrete:
-        return stats.bernoulli(p=self.params.p)
+    def distribution(self) -> scipy.stats.rv_discrete:
+        return scipy.stats.bernoulli(p=self.params.p)
 
 
 @SamplerRegistry.register(SamplerType.BERNOULLI_MIXTURE)
 class BernoulliMixtureSampler(TypeConversionMixin, Sampler[BernoulliMixtureSamplerParams]):
     def sample(self, num_samples: int) -> NumpyArray1dT:
-        return stats.bernoulli(p=self.params.p).rvs(size=num_samples) * getattr(stats, self.params.dist_name)(
-            **self.params.dist_params
-        ).rvs(size=num_samples)
+        return scipy.stats.bernoulli(p=self.params.p).rvs(size=num_samples) * getattr(
+            scipy.stats, self.params.dist_name
+        )(**self.params.dist_params).rvs(size=num_samples)
 
     def _validate(self) -> None:
         _validate_scipy_distribution(self.params.dist_name, self.params.dist_params)
@@ -292,29 +297,29 @@ class BernoulliMixtureSampler(TypeConversionMixin, Sampler[BernoulliMixtureSampl
 @SamplerRegistry.register(SamplerType.BINOMIAL)
 class BinomialSampler(TypeConversionMixin, ScipyStatsSampler[BinomialSamplerParams]):
     @property
-    def distribution(self) -> stats.rv_discrete:
-        return stats.binom(n=self.params.n, p=self.params.p)
+    def distribution(self) -> scipy.stats.rv_discrete:
+        return scipy.stats.binom(n=self.params.n, p=self.params.p)
 
 
 @SamplerRegistry.register(SamplerType.GAUSSIAN)
 class GaussianSampler(TypeConversionMixin, ScipyStatsSampler[GaussianSamplerParams]):
     @property
-    def distribution(self) -> stats.rv_continuous:
-        return stats.norm(loc=self.params.mean, scale=self.params.stddev)
+    def distribution(self) -> scipy.stats.rv_continuous:
+        return scipy.stats.norm(loc=self.params.mean, scale=self.params.stddev)
 
 
 @SamplerRegistry.register(SamplerType.POISSON)
 class PoissonSampler(TypeConversionMixin, ScipyStatsSampler[PoissonSamplerParams]):
     @property
-    def distribution(self) -> stats.rv_discrete:
-        return stats.poisson(mu=self.params.mean)
+    def distribution(self) -> scipy.stats.rv_discrete:
+        return scipy.stats.poisson(mu=self.params.mean)
 
 
 @SamplerRegistry.register(SamplerType.UNIFORM)
 class UniformSampler(TypeConversionMixin, ScipyStatsSampler[UniformSamplerParams]):
     @property
-    def distribution(self) -> stats.rv_continuous:
-        return stats.uniform(loc=self.params.low, scale=self.params.high - self.params.low)
+    def distribution(self) -> scipy.stats.rv_continuous:
+        return scipy.stats.uniform(loc=self.params.low, scale=self.params.high - self.params.low)
 
 
 ###################################################
@@ -328,14 +333,14 @@ def load_sampler(sampler_type: SamplerType, **params) -> DataSource:
 
 
 def _validate_scipy_distribution(dist_name: str, dist_params: dict) -> None:
-    if not hasattr(stats, dist_name):
+    if not hasattr(scipy.stats, dist_name):
         raise InvalidSamplerParamsError(f"Distribution {dist_name} not found in scipy.stats")
-    if not hasattr(getattr(stats, dist_name), "rvs"):
+    if not hasattr(getattr(scipy.stats, dist_name), "rvs"):
         raise InvalidSamplerParamsError(
             f"Distribution {dist_name} does not have a `rvs` method, which is required for sampling."
         )
     try:
-        getattr(stats, dist_name)(**dist_params)
+        getattr(scipy.stats, dist_name)(**dist_params)
     except Exception:
         raise InvalidSamplerParamsError(
             f"Distribution parameters {dist_params} are not a valid for distribution '{dist_name}'"

data_designer/engine/sampling_gen/entities/dataset_based_person_fields.py

@@ -10,8 +10,9 @@ This file contains all possible fields that:
 Do not add any other code or logic in this file.
 """
 
-REQUIRED_FIELDS = {"first_name", "last_name", "age", "locale"}
+from __future__ import annotations
 
+REQUIRED_FIELDS = {"first_name", "last_name", "age", "locale"}
 
 PII_FIELDS = [
     # Core demographic fields
@@ -52,7 +53,6 @@ PII_FIELDS = [
     "third_language",
 ]
 
-
 PERSONA_FIELDS = [
     # Core persona fields
     "persona",

data_designer/engine/sampling_gen/entities/email_address_utils.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
+
 import random
 import re
 from datetime import date

data_designer/engine/sampling_gen/entities/errors.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 data_designer.errors import DataDesignerError
 
 

data_designer/engine/sampling_gen/entities/national_id_utils.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
+
 import random
 from datetime import date
 

data_designer/engine/sampling_gen/entities/person.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
+
 import random
 from datetime import date, timedelta
 from typing import Any, Literal, TypeAlias

data_designer/engine/sampling_gen/entities/phone_number.py

@@ -1,12 +1,19 @@
 # SPDX-FileCopyrightText: Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 # SPDX-License-Identifier: Apache-2.0
 
+from __future__ import annotations
+
 import random
 from pathlib import Path
+from typing import TYPE_CHECKING
 
-import pandas as pd
 from pydantic import BaseModel, Field, field_validator
 
+from data_designer.lazy_heavy_imports import pd
+
+if TYPE_CHECKING:
+    import pandas as pd
+
 ZIP_AREA_CODE_DATA = pd.read_parquet(Path(__file__).parent / "assets" / "zip_area_code_map.parquet")
 ZIPCODE_AREA_CODE_MAP = dict(zip(ZIP_AREA_CODE_DATA["zipcode"], ZIP_AREA_CODE_DATA["area_code"]))
 ZIPCODE_POPULATION_MAP = dict(zip(ZIP_AREA_CODE_DATA["zipcode"], ZIP_AREA_CODE_DATA["count"]))

data_designer/engine/sampling_gen/errors.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 data_designer.engine.errors import DataDesignerError
 
 

data_designer/engine/sampling_gen/generator.py

@@ -6,18 +6,19 @@ from __future__ import annotations
 from collections.abc import Callable
 from typing import TYPE_CHECKING
 
-import networkx as nx
-import numpy as np
-import pandas as pd
-
 from data_designer.engine.sampling_gen.data_sources.base import RadomStateT
 from data_designer.engine.sampling_gen.errors import RejectionSamplingError
 from data_designer.engine.sampling_gen.jinja_utils import JinjaDataFrame
 from data_designer.engine.sampling_gen.people_gen import create_people_gen_resource
 from data_designer.engine.sampling_gen.schema import DataSchema
 from data_designer.engine.sampling_gen.utils import check_random_state
+from data_designer.lazy_heavy_imports import np, nx, pd
 
 if TYPE_CHECKING:
+    import networkx as nx
+    import numpy as np
+    import pandas as pd
+
     from data_designer.engine.dataset_builders.multi_column_configs import SamplerMultiColumnConfig
     from data_designer.engine.resources.managed_dataset_generator import ManagedDatasetGenerator
     from data_designer.engine.sampling_gen.column import ConditionalDataColumn

data_designer/engine/sampling_gen/jinja_utils.py

@@ -1,15 +1,19 @@
 # SPDX-FileCopyrightText: Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 # SPDX-License-Identifier: Apache-2.0
 
-import ast
-from typing import Any
+from __future__ import annotations
 
-import pandas as pd
+import ast
+from typing import TYPE_CHECKING, Any
 
 from data_designer.engine.processing.ginja.environment import (
     UserTemplateSandboxEnvironment,
     WithJinja2UserTemplateRendering,
 )
+from data_designer.lazy_heavy_imports import pd
+
+if TYPE_CHECKING:
+    import pandas as pd
 
 
 class JinjaDataFrame(WithJinja2UserTemplateRendering):

data_designer/engine/sampling_gen/people_gen.py

@@ -10,9 +10,6 @@ from collections.abc import Callable
 from copy import deepcopy
 from typing import TYPE_CHECKING, Any, TypeAlias
 
-import pandas as pd
-from faker import Faker
-
 from data_designer.config.utils.constants import DEFAULT_AGE_RANGE
 from data_designer.engine.resources.managed_dataset_generator import ManagedDatasetGenerator
 from data_designer.engine.sampling_gen.entities.dataset_based_person_fields import PERSONA_FIELDS, PII_FIELDS
@@ -22,12 +19,15 @@ from data_designer.engine.sampling_gen.entities.person import (
 )
 from data_designer.engine.sampling_gen.errors import ManagedDatasetGeneratorError
 from data_designer.engine.sampling_gen.person_constants import faker_constants
+from data_designer.lazy_heavy_imports import faker, pd
 
 if TYPE_CHECKING:
-    from data_designer.engine.sampling_gen.schema import DataSchema
+    import faker
+    import pandas as pd
 
+    from data_designer.engine.sampling_gen.schema import DataSchema
 
-EngineT: TypeAlias = Faker | ManagedDatasetGenerator
+EngineT: TypeAlias = faker.Faker | ManagedDatasetGenerator
 
 
 class PeopleGen(ABC):
@@ -46,7 +46,7 @@ class PeopleGen(ABC):
 
 class PeopleGenFaker(PeopleGen):
     @property
-    def _fake(self) -> Faker:
+    def _fake(self) -> faker.Faker:
         return self._engine
 
     def try_fake_else_none(self, attr_name: str, none_fill: Any | None = None) -> type:
@@ -193,7 +193,7 @@ def create_people_gen_resource(
         for params in [column.params, *list(column.conditional_params.values())]:
             if params.people_gen_key not in people_gen_resource:
                 people_gen_resource[params.people_gen_key] = PeopleGenFaker(
-                    engine=Faker(params.locale), locale=params.locale
+                    engine=faker.Faker(params.locale), locale=params.locale
                 )
 
     return people_gen_resource

data_designer/engine/sampling_gen/person_constants.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 typing import NamedTuple
 
 

data_designer/engine/sampling_gen/schema.py

@@ -4,8 +4,8 @@
 from __future__ import annotations
 
 from functools import cached_property
+from typing import TYPE_CHECKING
 
-import networkx as nx
 from pydantic import BaseModel, Field, field_validator, model_validator
 from typing_extensions import Self
 
@@ -14,6 +14,10 @@ from data_designer.config.sampler_constraints import ColumnConstraintT
 from data_designer.config.sampler_params import SamplerType
 from data_designer.engine.sampling_gen.column import ConditionalDataColumn
 from data_designer.engine.sampling_gen.constraints import ConstraintChecker, get_constraint_checker
+from data_designer.lazy_heavy_imports import nx
+
+if TYPE_CHECKING:
+    import networkx as nx
 
 
 class Dag(BaseModel):

data_designer/engine/sampling_gen/schema_builder.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 copy import deepcopy
 
 from data_designer.config.column_configs import SamplerColumnConfig

data_designer/engine/sampling_gen/utils.py

@@ -1,9 +1,15 @@
 # SPDX-FileCopyrightText: Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 # SPDX-License-Identifier: Apache-2.0
 
+from __future__ import annotations
+
 import numbers
+from typing import TYPE_CHECKING
+
+from data_designer.lazy_heavy_imports import np
 
-import numpy as np
+if TYPE_CHECKING:
+    import numpy as np
 
 
 def check_random_state(seed):

data_designer/engine/secret_resolver.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
+
 import json
 import logging
 import os

data_designer/engine/validation.py

@@ -19,7 +19,7 @@ from data_designer.config.processors import ProcessorConfigT, ProcessorType
 from data_designer.config.utils.constants import RICH_CONSOLE_THEME
 from data_designer.config.utils.misc import (
     can_run_data_designer_locally,
-    get_prompt_template_keywords,
+    extract_keywords_from_jinja2_template,
 )
 from data_designer.config.validator_params import ValidatorType
 from data_designer.engine.column_generators.utils.generator_classification import column_type_is_model_generated
@@ -302,7 +302,7 @@ def validate_schema_transform_processor(
     for processor_config in processor_configs:
         if processor_config.processor_type == ProcessorType.SCHEMA_TRANSFORM:
             for col, template in processor_config.template.items():
-                template_keywords = get_prompt_template_keywords(template)
+                template_keywords = extract_keywords_from_jinja2_template(template)
                 invalid_keywords = set(template_keywords) - all_column_names
                 if len(invalid_keywords) > 0:
                     invalid_keywords = ", ".join([f"'{k}'" for k in invalid_keywords])

data_designer/engine/validators/__init__.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 data_designer.engine.validators.base import BaseValidator, ValidationResult
 from data_designer.engine.validators.local_callable import LocalCallableValidator
 from data_designer.engine.validators.python import PythonValidator

data_designer/engine/validators/base.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 abc import ABC, abstractmethod
 from typing import Iterator
 

data_designer/engine/validators/local_callable.py

@@ -1,14 +1,19 @@
 # SPDX-FileCopyrightText: Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 # SPDX-License-Identifier: Apache-2.0
 
-import logging
+from __future__ import annotations
 
-import pandas as pd
+import logging
+from typing import TYPE_CHECKING
 
 from data_designer.config.validator_params import LocalCallableValidatorParams
 from data_designer.engine.errors import LocalCallableValidationError
 from data_designer.engine.processing.gsonschema.validators import validate
 from data_designer.engine.validators.base import BaseValidator, ValidationOutput, ValidationResult
+from data_designer.lazy_heavy_imports import pd
+
+if TYPE_CHECKING:
+    import pandas as pd
 
 logger = logging.getLogger(__name__)
 

data_designer/engine/validators/python.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
+
 import ast
 import json
 import logging
@@ -8,14 +10,18 @@ import subprocess
 import tempfile
 from collections import defaultdict
 from pathlib import Path
+from typing import TYPE_CHECKING
 from uuid import uuid4
 
-import pandas as pd
 from pydantic import BaseModel
 from ruff.__main__ import find_ruff_bin
 
 from data_designer.config.validator_params import CodeValidatorParams
 from data_designer.engine.validators.base import BaseValidator, ValidationOutput, ValidationResult
+from data_designer.lazy_heavy_imports import pd
+
+if TYPE_CHECKING:
+    import pandas as pd
 
 logger = logging.getLogger(__name__)
 

data_designer/engine/validators/remote.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
+
 import logging
+from typing import TYPE_CHECKING
 
-import httpx
 from httpx_retries import Retry, RetryTransport
 
 from data_designer.config.validator_params import RemoteValidatorParams
@@ -11,6 +13,10 @@ from data_designer.engine.errors import RemoteValidationSchemaError
 from data_designer.engine.processing.gsonschema.exceptions import JSONSchemaValidationError
 from data_designer.engine.processing.gsonschema.validators import validate
 from data_designer.engine.validators.base import BaseValidator, ValidationResult
+from data_designer.lazy_heavy_imports import httpx
+
+if TYPE_CHECKING:
+    import httpx
 
 logger = logging.getLogger(__name__)
 

data_designer/engine/validators/sql.py

@@ -1,15 +1,20 @@
 # SPDX-FileCopyrightText: Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 # SPDX-License-Identifier: Apache-2.0
 
+from __future__ import annotations
+
 import logging
 import re
-
-import pandas as pd
-import sqlfluff
+from typing import TYPE_CHECKING
 
 from data_designer.config.utils.code_lang import CodeLang
 from data_designer.config.validator_params import CodeValidatorParams
 from data_designer.engine.validators.base import BaseValidator, ValidationOutput, ValidationResult
+from data_designer.lazy_heavy_imports import pd, sqlfluff
+
+if TYPE_CHECKING:
+    import pandas as pd
+    import sqlfluff
 
 sqlfluff_logger = logging.getLogger("sqlfluff")
 sqlfluff_logger.setLevel(logging.WARNING)

data_designer/errors.py

@@ -1,5 +1,7 @@
 # SPDX-FileCopyrightText: Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 # SPDX-License-Identifier: Apache-2.0
 
+from __future__ import annotations
+
 
 class DataDesignerError(Exception): ...

data_designer/essentials/__init__.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 data_designer.config.default_model_settings import resolve_seed_default_model_settings
 from data_designer.config.exports import *  # noqa: F403
 from data_designer.config.run_config import RunConfig