skip mypy, fix linting

llama_stack/providers/remote/post_training/nvidia/post_training.py

@@ -67,7 +67,7 @@ class NvidiaPostTrainingAdapter(PostTraining, ModelRegistryHelper):
         self.customizer_url = config.customizer_url
 
         if not self.customizer_url:
-            warnings.warn("Customizer URL is not set, using default value: http://nemo.test")
+            warnings.warn("Customizer URL is not set, using default value: http://nemo.test", stacklevel=2)
             self.customizer_url = "http://nemo.test"
 
     async def _make_request(
@@ -122,8 +122,16 @@ class NvidiaPostTrainingAdapter(PostTraining, ModelRegistryHelper):
             mapped_status = STATUS_MAPPING.get(job_status, "unknown")
 
             # Convert string timestamps to datetime objects
-            created_at = datetime.fromisoformat(job.pop("created_at")) if "created_at" in job else datetime.now()
-            updated_at = datetime.fromisoformat(job.pop("updated_at")) if "updated_at" in job else datetime.now()
+            created_at = (
+                datetime.fromisoformat(job.pop("created_at"))
+                if "created_at" in job
+                else datetime.now(tz=datetime.timezone.utc)
+            )
+            updated_at = (
+                datetime.fromisoformat(job.pop("updated_at"))
+                if "updated_at" in job
+                else datetime.now(tz=datetime.timezone.utc)
+            )
 
             # Create NvidiaPostTrainingJob instance
             jobs.append(
@@ -302,7 +310,10 @@ class NvidiaPostTrainingAdapter(PostTraining, ModelRegistryHelper):
             unsupported_method_params.append("logger_config")
 
         if unsupported_method_params:
-            warnings.warn(f"Parameters: {', '.join(unsupported_method_params)} are not supported and will be ignored")
+            warnings.warn(
+                f"Parameters: {', '.join(unsupported_method_params)} are not supported and will be ignored",
+                stacklevel=2,
+            )
 
         # Define all supported parameters
         supported_params = {

llama_stack/providers/remote/post_training/nvidia/utils.py

@@ -22,7 +22,9 @@ def warn_unsupported_params(config_dict: Any, supported_keys: Set[str], config_n
     keys = set(config_dict.__annotations__.keys()) if isinstance(config_dict, BaseModel) else config_dict.keys()
     unsupported_params = [k for k in keys if k not in supported_keys]
     if unsupported_params:
-        warnings.warn(f"Parameters: {unsupported_params} in `{config_name}` not supported and will be ignored.")
+        warnings.warn(
+            f"Parameters: {unsupported_params} in `{config_name}` not supported and will be ignored.", stacklevel=2
+        )
 
 
 def validate_training_params(
@@ -49,7 +51,9 @@ def validate_training_params(
                 unsupported_params.append(key)
 
     if unsupported_params:
-        warnings.warn(f"Parameters: {unsupported_params} in `{config_name}` are not supported and will be ignored.")
+        warnings.warn(
+            f"Parameters: {unsupported_params} in `{config_name}` are not supported and will be ignored.", stacklevel=2
+        )
 
 
 # ToDo: implement post health checks for customizer are enabled

pyproject.toml

@@ -272,6 +272,7 @@ exclude = [
     "^llama_stack/providers/remote/tool_runtime/model_context_protocol/",
     "^llama_stack/providers/remote/tool_runtime/tavily_search/",
     "^llama_stack/providers/remote/tool_runtime/wolfram_alpha/",
+    "^llama_stack/providers/remote/post_training/nvidia/",
     "^llama_stack/providers/remote/vector_io/chroma/",
     "^llama_stack/providers/remote/vector_io/milvus/",
     "^llama_stack/providers/remote/vector_io/pgvector/",