support qcodes 0.37

QCoDeS · Jan 30, 2023 · 3d5b983 · 3d5b983
1 parent 772f832
commit 3d5b983
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/qcodes_contrib_drivers/tests/test_Keysight_M3201A.py b/qcodes_contrib_drivers/tests/test_Keysight_M3201A.py
@@ -1,4 +1,8 @@
-from qcodes.instrument_drivers.test import DriverTestCase
+try:
+    from qcodes.tests.driver_test_case import DriverTestCase
+except ImportError:
+    # qcodes < 0.37
+    from qcodes.instrument_drivers.test import DriverTestCase
 import unittest
 
 try: