conditionally import celery_app

meta_edc/__init__.py

@@ -1,14 +1,18 @@
 import os
+import sys
 from importlib.metadata import PackageNotFoundError, version
 
-from .celery.debug import app as celery_debug
-from .celery.live import app as celery_live
-from .celery.uat import app as celery_uat
+if "meta_edc.celery.live:app" in sys.argv:
+    from .celery.live import app as celery_app
+elif "meta_edc.celery.uat:app" in sys.argv:
+    from .celery.uat import app as celery_app
+else:
+    from .celery.debug import app as celery_app
 
 try:
     __version__ = version(os.getcwd().split(os.sep)[-1])
 except PackageNotFoundError:
     __version__ = None
 
 
-__all__ = ["celery_live", "celery_uat", "celery_debug", "__version__"]
+__all__ = ["celery_app", "__version__"]