🔖 0.12.4 (#186)

* Modify sys.argv before the module is loaded in `utils.load_pipeline()`

* Fix linting

* 🔖 0.12.4

docs/CHANGELOG.md

@@ -1,5 +1,9 @@
 # Change Log
 
+## 0.12.4
+
+- Modify sys.argv before the module is loaded in `utils.load_pipeline()`
+
 ## 0.12.3
 
 - Change cli_args to argv0 and argv1p for utils.load_pipeline

pipen/utils.py

@@ -665,40 +665,43 @@ async def load_pipeline(
     from .proc import Proc
     from .procgroup import ProcGroup
 
-    if isinstance(obj, str):
-        obj = _get_obj_from_spec(obj)
-        if (
-            not isinstance(obj, type)
-            or not issubclass(obj, (Pipen, Proc, ProcGroup))
-        ):
-            raise TypeError(
-                f"Expected a Pipen, Proc or ProcGroup class, got {type(obj)}"
-            )
-
-    pipeline = obj
-    if isinstance(obj, type) and issubclass(obj, Proc):
-        kwargs.setdefault("name", f"{obj.name}Pipeline")
-        pipeline = Pipen(**kwargs).set_starts(obj)
-
-    if isinstance(obj, type) and issubclass(obj, ProcGroup):
-        pipeline = obj().as_pipen(**kwargs)
-
-    if isinstance(obj, type) and issubclass(obj, Pipen):
-        # Avoid "pipeline" to be used as pipeline name by varname
-        (pipeline, ) = (obj(**kwargs), )
-
-    if not isinstance(pipeline, Pipen):
-        raise TypeError(
-            f"Expected a Pipen, Proc or ProcGroup class, got {type(pipeline)}"
-        )
-
     old_argv = sys.argv
     if argv0 is None:
         argv0 = LOADING_ARGV0
     if argv1p is None:
         argv1p = sys.argv[1:]
     sys.argv = [argv0] + list(argv1p)
+
     try:
+        if isinstance(obj, str):
+            obj = _get_obj_from_spec(obj)
+            if (
+                not isinstance(obj, type)
+                or not issubclass(obj, (Pipen, Proc, ProcGroup))
+            ):
+                raise TypeError(
+                    "Expected a Pipen, Proc or ProcGroup class, "
+                    f"got {type(obj)}"
+                )
+
+        pipeline = obj
+        if isinstance(obj, type) and issubclass(obj, Proc):
+            kwargs.setdefault("name", f"{obj.name}Pipeline")
+            pipeline = Pipen(**kwargs).set_starts(obj)
+
+        if isinstance(obj, type) and issubclass(obj, ProcGroup):
+            pipeline = obj().as_pipen(**kwargs)
+
+        if isinstance(obj, type) and issubclass(obj, Pipen):
+            # Avoid "pipeline" to be used as pipeline name by varname
+            (pipeline, ) = (obj(**kwargs), )
+
+        if not isinstance(pipeline, Pipen):
+            raise TypeError(
+                "Expected a Pipen, Proc or ProcGroup class, "
+                f"got {type(pipeline)}"
+            )
+
         # Initialize the pipeline so that the arguments definied by
         # other plugins (i.e. pipen-args) to take in place.
         pipeline.workdir = Path(pipeline.config.workdir).joinpath(
@@ -717,6 +720,10 @@ def is_loading_pipeline() -> bool:
     """Check if we are loading the pipeline. Works only when
     `argv0` is "@pipen" while loading the pipeline.
 
+    Note if you are using this function at compile time, make
+    sure you load your pipeline using the string form (`part1:part2`)
+    See more with `load_pipline()`.
+
     Returns:
         True if we are loading the pipeline (argv[0] == "@pipen"),
         otherwise False

pipen/version.py

@@ -1,3 +1,3 @@
 """Provide version of pipen"""
 
-__version__ = "0.12.3"
+__version__ = "0.12.4"

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.masonry.api"
 
 [tool.poetry]
 name = "pipen"
-version = "0.12.3"
+version = "0.12.4"
 description = "A pipeline framework for python"
 authors = [ "pwwang <[email protected]>",]
 license = "MIT"