Increase similarity to old version

src/instamatic/TEMController/TEMController.py

@@ -1,3 +1,3 @@
 from __future__ import annotations
 
-from instamatic.controller import TEMController
+from instamatic.controller import TEMController, get_instance, initialize

src/instamatic/TEMController/__init__.py

@@ -3,21 +3,13 @@
 
 import warnings
 
-from instamatic.utils.deprecated import VisibleDeprecationWarning, deprecated
+from instamatic.utils.deprecated import VisibleDeprecationWarning
 
 warnings.warn(
     'The `TEMController` module is deprecated since version 2.0.6. Use the `controller`-module instead',
     VisibleDeprecationWarning,
     stacklevel=2,
 )
 
-
-from instamatic.controller import get_instance, initialize
-from instamatic.microscope.base import MicroscopeBase
-
-
-@deprecated(since='2.0.6', alternative='instamatic.microscope.get_microscope')
-def Microscope(name: str = None, use_server: bool = False) -> MicroscopeBase:
-    from instamatic.microscope import get_microscope
-
-    return get_microscope(name=name, use_server=use_server)
+from .microscope import Microscope
+from .TEMController import get_instance, initialize

src/instamatic/TEMController/microscope.py

@@ -1,10 +1,20 @@
 from __future__ import annotations
 
+from instamatic.microscope.base import MicroscopeBase
 from instamatic.utils.deprecated import deprecated
 
+__all__ = ['Microscope', 'get_tem']
+
 
 @deprecated(since='2.0.6', alternative='instamatic.microscope.get_microscope_class')
 def get_tem(interface: str) -> 'type[MicroscopeBase]':
     from instamatic.microscope import get_microscope_class
 
     return get_microscope_class(interface=interface)
+
+
+@deprecated(since='2.0.6', alternative='instamatic.microscope.get_microscope')
+def Microscope(name: str = None, use_server: bool = False) -> MicroscopeBase:
+    from instamatic.microscope import get_microscope
+
+    return get_microscope(name=name, use_server=use_server)