fix: added LogisticRegression to all

NillionNetwork · Jun 13, 2024 · d9fa1e4 · d9fa1e4
1 parent ddf27b4
commit d9fa1e4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nada_ai/linear_model/linear_regression.py b/nada_ai/linear_model/linear_regression.py
@@ -4,7 +4,7 @@
 from nada_ai.nn.module import Module
 from nada_ai.nn.parameter import Parameter
 
-__all__ = ["LinearRegression"]
+__all__ = ["LinearRegression", "LogisticRegression"]
 
 
 class LinearRegression(Module):