From 9b3dbb286e5cff004973b2e82c22c48d588c0ad1 Mon Sep 17 00:00:00 2001
From: Patrick <>
Date: Sun, 11 Feb 2024 09:00:50 +0100
Subject: [PATCH] expose classes and methods for convenience

---
 src/qutip_qoc/__init__.py      | 4 +++-
 src/qutip_qoc/time_interval.py | 3 +++
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/src/qutip_qoc/__init__.py b/src/qutip_qoc/__init__.py
index ec147f1..d6fc86e 100644
--- a/src/qutip_qoc/__init__.py
+++ b/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"]
diff --git a/src/qutip_qoc/time_interval.py b/src/qutip_qoc/time_interval.py
index ba7749d..d929c17 100644
--- a/src/qutip_qoc/time_interval.py
+++ b/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: