Traceback (most recent call last): File "/home/fynnsu/repro/.venv/lib/python3.10/site-packages/pydantic/_internal/_generate_schema.py", line 957, in _resolve_forward_ref obj = _typing_extra.eval_type_backport(obj, *self._types_namespace) File "/home/fynnsu/repro/.venv/lib/python3.10/site-packages/pydantic/_internal/_typing_extra.py", line 455, in eval_type_backport return _eval_type_backport(value, globalns, localns, type_params) File "/home/fynnsu/repro/.venv/lib/python3.10/site-packages/pydantic/_internal/_typing_extra.py", line 492, in _eval_type_backport return _eval_type(value, globalns, localns, type_params) File "/home/fynnsu/repro/.venv/lib/python3.10/site-packages/pydantic/_internal/_typing_extra.py", line 545, in _eval_type return typing._eval_type( # type: ignore File "/home/fynnsu/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/typing.py", line 327, in _eval_type return t._evaluate(globalns, localns, recursive_guard) File "/home/fynnsu/.local/share/uv/python/cpython-3.10.18-linux-x86_64-gnu/lib/python3.10/typing.py", line 694, in _evaluate eval(self.forward_code, globalns, localns), File "<string>", line 1, in <module> NameError: name 'torch' is not defined

The above exception was the direct cause of the following exception:

Traceback (most recent call last): File "/home/fynnsu/repro/transformers_pydantic.py", line 12, in <module> MyModelConfig.model_rebuild(force=True) File "/home/fynnsu/repro/.venv/lib/python3.10/site-packages/pydantic/main.py", line 668, in model_rebuild return _model_construction.complete_model_class( File "/home/fynnsu/repro/.venv/lib/python3.10/site-packages/pydantic/_internal/_model_construction.py", line 648, in complete_model_class schema = gen_schema.generate_schema(cls) File "/home/fynnsu/repro/.venv/lib/python3.10/site-packages/pydantic/_internal/_generate_schema.py", line 729, in generate_schema schema = self._generate_schema_inner(obj) File "/home/fynnsu/repro/.venv/lib/python3.10/site-packages/pydantic/_internal/_generate_schema.py", line 1023, in _generate_schema_inner return self._model_schema(obj) File "/home/fynnsu/repro/.venv/lib/python3.10/site-packages/pydantic/_internal/_generate_schema.py", line 856, in _model_schema {k: self._generate_md_field_schema(k, v, decorators) for k, v in fields.items()}, File "/home/fynnsu/repro/.venv/lib/python3.10/site-packages/pydantic/_internal/_generate_schema.py", line 856, in <dictcomp> {k: self._generate_md_field_schema(k, v, decorators) for k, v in fields.items()}, File "/home/fynnsu/repro/.venv/lib/python3.10/site-packages/pydantic/_internal/_generate_schema.py", line 1228, in _generate_md_field_schema schema, metadata = self._common_field_schema(name, field_info, decorators) File "/home/fynnsu/repro/.venv/lib/python3.10/site-packages/pydantic/_internal/_generate_schema.py", line 1282, in _common_field_schema schema = self._apply_annotations( File "/home/fynnsu/repro/.venv/lib/python3.10/site-packages/pydantic/_internal/_generate_schema.py", line 2227, in _apply_annotations schema = get_inner_schema(source_type) File "/home/fynnsu/repro/.venv/lib/python3.10/site-packages/pydantic/_internal/_schema_generation_shared.py", line 83, in call schema = self._handler(source_type) File "/home/fynnsu/repro/.venv/lib/python3.10/site-packages/pydantic/_internal/_generate_schema.py", line 2206, in inner_handler schema = self._generate_schema_inner(obj) File "/home/fynnsu/repro/.venv/lib/python3.10/site-packages/pydantic/_internal/_generate_schema.py", line 1028, in _generate_schema_inner return self.match_type(obj) File "/home/fynnsu/repro/.venv/lib/python3.10/site-packages/pydantic/_internal/_generate_schema.py", line 1140, in match_type return self._dataclass_schema(obj, None) # pyright: ignore[reportArgumentType] File "/home/fynnsu/repro/.venv/lib/python3.10/site-packages/pydantic/_internal/_generate_schema.py", line 1897, in _dataclass_schema args = sorted( File "/home/fynnsu/repro/.venv/lib/python3.10/site-packages/pydantic/_internal/_generate_schema.py", line 1898, in <genexpr> (self._generate_dc_field_schema(k, v, decorators) for k, v in fields.items()), File "/home/fynnsu/repro/.venv/lib/python3.10/site-packages/pydantic/_internal/_generate_schema.py", line 1246, in _generate_dc_field_schema schema, metadata = self._common_field_schema(name, field_info, decorators) File "/home/fynnsu/repro/.venv/lib/python3.10/site-packages/pydantic/_internal/_generate_schema.py", line 1282, in _common_field_schema schema = self._apply_annotations( File "/home/fynnsu/repro/.venv/lib/python3.10/site-packages/pydantic/_internal/_generate_schema.py", line 2227, in _apply_annotations schema = get_inner_schema(source_type) File "/home/fynnsu/repro/.venv/lib/python3.10/site-packages/pydantic/_internal/_schema_generation_shared.py", line 83, in call schema = self._handler(source_type) File "/home/fynnsu/repro/.venv/lib/python3.10/site-packages/pydantic/_internal/_generate_schema.py", line 2206, in inner_handler schema = self._generate_schema_inner(obj) File "/home/fynnsu/repro/.venv/lib/python3.10/site-packages/pydantic/_internal/_generate_schema.py", line 1028, in _generate_schema_inner return self.match_type(obj) File "/home/fynnsu/repro/.venv/lib/python3.10/site-packages/pydantic/_internal/_generate_schema.py", line 1144, in match_type return self._match_generic_type(obj, origin) File "/home/fynnsu/repro/.venv/lib/python3.10/site-packages/pydantic/_internal/_generate_schema.py", line 1167, in _match_generic_type return self._union_schema(obj) File "/home/fynnsu/repro/.venv/lib/python3.10/site-packages/pydantic/_internal/_generate_schema.py", line 1320, in _union_schema args = self._get_args_resolving_forward_refs(union_type, required=True) File "/home/fynnsu/repro/.venv/lib/python3.10/site-packages/pydantic/_internal/_generate_schema.py", line 982, in _get_args_resolving_forward_refs args = tuple(self._resolve_forward_ref(a) if isinstance(a, ForwardRef) else a for a in args) File "/home/fynnsu/repro/.venv/lib/python3.10/site-packages/pydantic/_internal/_generate_schema.py", line 982, in <genexpr> args = tuple(self._resolve_forward_ref(a) if isinstance(a, ForwardRef) else a for a in args) File "/home/fynnsu/repro/.venv/lib/python3.10/site-packages/pydantic/_internal/_generate_schema.py", line 959, in _resolve_forward_ref raise PydanticUndefinedAnnotation.from_name_error(e) from e pydantic.errors.PydanticUndefinedAnnotation: name 'torch' is not defined

For further information visit https://errors.pydantic.dev/2.12/u/undefined-annotation

Fix Plan

To fix the issue, you need to import the torch module to make it available in the local namespace.

Here are the steps:

• Import the torch module at the beginning of your script.
• Ensure that the torch module is installed in your environment.

Code Changes

import torch
from pydantic import BaseModel, ConfigDict, Field
from transformers import PretrainedConfig

class MyModelConfig(BaseModel):
    model_config = ConfigDict(arbitrary_types_allowed=True)
    sub_config: PretrainedConfig = Field(
        description="Configuration for the sub_config",
    )

# Note: model_rebuild is not a valid method for Pydantic models
# If you are trying to rebuild the model, you might need to create a new instance
my_model_config = MyModelConfig(model_config={}, sub_config=PretrainedConfig())

Verification

To verify that the fix worked, you can try to create an instance of MyModelConfig without any errors.

try:
    my_model_config = MyModelConfig(model_config={}, sub_config=PretrainedConfig())
    print("MyModelConfig instance created successfully")
except Exception as e:
    print(f"Error creating MyModelConfig instance: {e}")

Extra Tips

• Make sure to install the torch module in your environment using pip install torch or uv pip install torch.
• If you are using a virtual environment, ensure that the torch module is installed in the correct environment.
• The model_rebuild method is not a valid method for Pydantic models. If you need to rebuild the model, you might need to create a new instance.