Add train particle classifier to __all__

Co-authored-by: Noah Biederbeck <[email protected]>
cta-observatory · Oct 27, 2022 · 692f81a · 692f81a
1 parent 6036825
commit 692f81a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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"]