expose classes and methods for convenience

src/qutip_qoc/__init__.py

@@ -1,5 +1,7 @@
 from .version import version as __version__  # noqa
 
 from qutip_qoc.time_interval import TimeInterval
+from qutip_qoc.objective import Objective
+from qutip_qoc.optimize import optimize_pulses
 
-__all__ = ["TimeInterval"]
+__all__ = ["TimeInterval", "Objective", "optimize_pulses"]

src/qutip_qoc/time_interval.py

@@ -35,6 +35,9 @@ def __init__(self, tslots=None, evo_time=None, n_tslots=100, tdiffs=None):
         self._n_tslots = n_tslots
         self._tdiffs = tdiffs
 
+    def __call__(self):
+        return self.tslots
+
     @property
     def tslots(self):
         if self._tslots is None: