update init

Author: DNXie

src/forge/observability/__init__.py

@@ -23,6 +23,7 @@
     MetricCollector,
     MinAccumulator,
     RandomRatioFilter,
+    record_episode_sample,
     record_metric,
     Reduce,
     reduce_metrics_states,
@@ -38,6 +39,7 @@
 __all__ = [
     # Main API functions
     "record_metric",
+    "record_episode_sample",
     "reduce_metrics_states",
     "get_actor_name_with_rank",
     "get_logger_backend_class",