Merge pull request #45 from QCoDeS/fix_import

Fix import
QCoDeS · Sep 14, 2023 · 4c4e584 · 4c4e584
2 parents 9afc0b0 + 111825e
commit 4c4e584
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/qcodes_loop/__init__.py b/src/qcodes_loop/__init__.py
@@ -6,7 +6,7 @@
 
 
 try:
-    _register_magic = qcodes.config.core.get('register_magic', False)
+    _register_magic = qcodes.config.core.get("register_magic", False)
     if _register_magic is not False:
         # get_ipython is part of the public api but IPython does
         # not use __all__ to mark this
@@ -15,7 +15,7 @@
         # Check if we are in IPython
         ip = get_ipython()
         if ip is not None:
-            from qcodes.utils.magic import register_magic_class
+            from qcodes_loop.utils.magic import register_magic_class
 
             register_magic_class(magic_commands=_register_magic)
 except ImportError: