import Device

stack-of-tasks · Jul 27, 2022 · 461fbb6 · 461fbb6
1 parent 016cea4
commit 461fbb6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/dynamic_graph/sot/core/__init__.py b/src/dynamic_graph/sot/core/__init__.py
@@ -1,2 +1,2 @@
 from . import wrap  # noqa
-from .wrap import Flags  # noqa
+from .wrap import Device, Flags  # noqa