fix: mute unsupported field attribute warning on startup

app.py

@@ -31,18 +31,19 @@
 import sys
 import warnings
 
-from nemo_automodel.cli.app import main
-
 # Pydantic v2 emits UnsupportedFieldAttributeWarning for Field(repr=...) /
 # Field(frozen=...) used inside 3.12-style `type` aliases in third-party libs.
 # There is nothing actionable for us here, so silence them globally.
+# Must run before any import that triggers pydantic schema generation.
 try:
     from pydantic.warnings import UnsupportedFieldAttributeWarning
 
     warnings.filterwarnings("ignore", category=UnsupportedFieldAttributeWarning)
 except ImportError:
     pass
 
+from nemo_automodel.cli.app import main
+
 logger = logging.getLogger(__name__)
 
 if __name__ == "__main__":

nemo_automodel/__init__.py

@@ -16,9 +16,21 @@
 import importlib.abc
 import importlib.machinery
 import sys
+import warnings
 from types import ModuleType
 from typing import Any
 
+# Pydantic v2 emits UnsupportedFieldAttributeWarning for Field(repr=...) /
+# Field(frozen=...) used inside 3.12-style `type` aliases in third-party libs.
+# Suppress early so any later import that triggers pydantic schema generation
+# (e.g. transformers, huggingface_hub) won't emit these warnings.
+try:
+    from pydantic.warnings import UnsupportedFieldAttributeWarning
+
+    warnings.filterwarnings("ignore", category=UnsupportedFieldAttributeWarning)
+except ImportError:
+    pass
+
 from .package_info import __package_name__, __version__
 
 # Keep the base package import lightweight.

nemo_automodel/recipes/llm/train_ft.py

@@ -14,6 +14,17 @@
 
 from __future__ import annotations
 
+import warnings
+
+# Suppress pydantic v2 UnsupportedFieldAttributeWarning before heavy imports
+# (transformers, huggingface_hub) trigger schema generation.
+try:
+    from pydantic.warnings import UnsupportedFieldAttributeWarning
+
+    warnings.filterwarnings("ignore", category=UnsupportedFieldAttributeWarning)
+except ImportError:
+    pass
+
 import inspect
 import logging
 import pathlib

nemo_automodel/recipes/vlm/finetune.py

@@ -14,6 +14,17 @@
 
 from __future__ import annotations
 
+import warnings
+
+# Suppress pydantic v2 UnsupportedFieldAttributeWarning before heavy imports
+# (transformers, huggingface_hub) trigger schema generation.
+try:
+    from pydantic.warnings import UnsupportedFieldAttributeWarning
+
+    warnings.filterwarnings("ignore", category=UnsupportedFieldAttributeWarning)
+except ImportError:
+    pass
+
 import logging
 import pathlib
 import time