add some functions to be easily import-able

pymf/__init__.py

@@ -7,8 +7,18 @@ except ImportError:
     __version_tuple__ = (0, 0, "unknown", "unknown")
 
 from .mf import construct_density_matrix
+from .solvers import solver
+from .model import Model
+from .tb.tb import add_tb, scale_tb
+from .tb.utils import generate_guess
+
 
 __all__ = [
+    "solver",
+    "Model",
+    "add_tb",
+    "scale_tb",
+    "generate_guess",
     "construct_density_matrix",
     "__version__",
     "__version_tuple__",