From 692f81a0c91a357aa36bb05e8242ca49454846a5 Mon Sep 17 00:00:00 2001
From: Lukas Nickel <lukas.nickel@tu-dortmund.de>
Date: Mon, 27 Jun 2022 16:49:31 +0200
Subject: [PATCH] Add train particle classifier to __all__

Co-authored-by: Noah Biederbeck <noah.biederbeck@tu-dortmund.de>
---
 ctapipe/ml/tools/__init__.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/ctapipe/ml/tools/__init__.py b/ctapipe/ml/tools/__init__.py
index 9150e3327eb..46c4f672467 100644
--- a/ctapipe/ml/tools/__init__.py
+++ b/ctapipe/ml/tools/__init__.py
@@ -1,3 +1,4 @@
 from .train_energy_regressor import TrainEnergyRegressor
+from .train_particle_classifier import TrainParticleIdClassifier
 
-__all__ = ["TrainEnergyRegressor"]
+__all__ = ["TrainEnergyRegressor", "TrainParticleIdClassifier"]