feat: observe decorators

src/judgeval/trainer/fireworks_trainer.py

@@ -1,7 +1,7 @@
 import asyncio
 import json
 from typing import Optional, Callable, Any, List, Union, Dict
-from fireworks import Dataset  # type: ignore[import-not-found]
+from fireworks import Dataset  # type: ignore[import-untyped]
 from .config import TrainerConfig, ModelConfig
 from .base_trainer import BaseTrainer
 from .trainable_model import TrainableModel

src/judgeval/trainer/trainable_model.py

@@ -1,5 +1,5 @@
 import time
-from fireworks import LLM  # type: ignore[import-not-found]
+from fireworks import LLM  # type: ignore[import-untyped]
 from .config import TrainerConfig, ModelConfig
 from typing import Optional, Dict, Any, Callable
 from .console import _model_spinner_progress, _print_model_progress