feat: observe decorators

src/judgeval/trainer/fireworks_trainer.py

@@ -1,7 +1,14 @@
 import asyncio
 import json
 from typing import Optional, Callable, Any, List, Union, Dict
-from fireworks import Dataset  # type: ignore[import-not-found,import-untyped]
+
+try:
+    from fireworks import Dataset  # type: ignore[import-not-found]
+except ImportError as e:
+    raise ImportError(
+        "Fireworks is not installed. Please install it with 'pip install fireworks'"
+    ) from e
+
 from .config import TrainerConfig, ModelConfig
 from .base_trainer import BaseTrainer
 from .trainable_model import TrainableModel

src/judgeval/trainer/trainable_model.py

@@ -1,7 +1,14 @@
 import time
-from fireworks import LLM  # type: ignore[import-not-found,import-untyped]
-from .config import TrainerConfig, ModelConfig
 from typing import Optional, Dict, Any, Callable
+
+try:
+    from fireworks import LLM  # type: ignore[import-not-found]
+except ImportError as e:
+    raise ImportError(
+        "Fireworks is not installed. Please install it with 'pip install fireworks'"
+    ) from e
+
+from .config import TrainerConfig, ModelConfig
 from .console import _model_spinner_progress, _print_model_progress
 from judgeval.exceptions import JudgmentRuntimeError