From ddcd84c740520163dbd4337437e4e18a201ae224 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?C=C3=A9dric=20Deltheil?= <cedric@deltheil.me>
Date: Wed, 29 May 2024 07:40:18 +0000
Subject: [PATCH] training_utils: export neptune config and mixin

Like W&B. With this, #371 would have broken `test_import_training_utils`.

Follow up of #371 and #372.
---
 src/refiners/training_utils/__init__.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/refiners/training_utils/__init__.py b/src/refiners/training_utils/__init__.py
index f39b762..f27bb52 100644
--- a/src/refiners/training_utils/__init__.py
+++ b/src/refiners/training_utils/__init__.py
@@ -41,6 +41,7 @@ from refiners.training_utils.config import (
     Optimizers,
     TrainingConfig,
 )
+from refiners.training_utils.neptune import NeptuneConfig, NeptuneMixin
 from refiners.training_utils.trainer import Trainer, register_callback, register_model
 from refiners.training_utils.wandb import WandbConfig, WandbMixin
 
@@ -54,6 +55,8 @@ __all__ = [
     "CallbackConfig",
     "WandbMixin",
     "WandbConfig",
+    "NeptuneMixin",
+    "NeptuneConfig",
     "LRSchedulerConfig",
     "OptimizerConfig",
     "TrainingConfig",