add backward compatibility with dflow (#414)

deepmodeling · Nov 17, 2023 · 038ced7 · 038ced7
1 parent ec84d6e
commit 038ced7
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 0 deletions.
diff --git a/dpdispatcher/dpcloudserver/__init__.py b/dpdispatcher/dpcloudserver/__init__.py
diff --git a/dpdispatcher/dpcloudserver/client.py b/dpdispatcher/dpcloudserver/client.py
@@ -0,0 +1,7 @@
+"""Provide backward compatbility with dflow."""
+
+from dpdispatcher.utils.dpcloudserver.client import RequestInfoException
+
+__all__ = [
+    "RequestInfoException",
+]
diff --git a/tests/context.py b/tests/context.py
@@ -11,6 +11,9 @@
 from dpdispatcher.contexts.lazy_local_context import LazyLocalContext  # noqa: F401
 from dpdispatcher.contexts.local_context import LocalContext  # noqa: F401
 from dpdispatcher.contexts.ssh_context import SSHContext, SSHSession  # noqa: F401
+
+# test backward compatibility with dflow
+from dpdispatcher.dpcloudserver.client import RequestInfoException as _  # noqa: F401
 from dpdispatcher.entrypoints.submission import handle_submission  # noqa: F401
 from dpdispatcher.machine import Machine  # noqa: F401
 from dpdispatcher.machines.distributed_shell import DistributedShell  # noqa: F401