refactor: added more __init__s

Author: Michal-Novomestsky

pymc_extras/inference/INLA/__init__.py

@@ -0,0 +1,3 @@
+from pymc_extras.inference.INLA.inla import fit_INLA
+
+__all__ = ["fit_INLA"]

pymc_extras/inference/fit.py

@@ -37,7 +37,7 @@ def fit(method: str, **kwargs) -> az.InferenceData:
         return fit_pathfinder(**kwargs)
 
     elif method == "laplace":
-        from pymc_extras.inference.laplace_approx.laplace import fit_laplace
+        from pymc_extras.inference.laplace_approx import fit_laplace
 
         return fit_laplace(**kwargs)
 

pymc_extras/inference/laplace_approx/__init__.py

@@ -0,0 +1,3 @@
+from pymc_extras.inference.laplace_approx.laplace import fit_laplace
+
+__all__ = ["fit_laplace"]