fix(components): enable desk for input artifact

Signed-off-by: weiwee <wbwmat@gmail.com>
FederatedAI · Jan 10, 2023 · 48febad · 48febad
1 parent 7478f15
commit 48febad
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/python/fate/components/cpn.py b/python/fate/components/cpn.py
@@ -188,7 +188,11 @@ def _dict(self):
                 )
             elif annotated == InputAnnotated:
                 input_artifacts[artifact.name] = ArtifactSpec(
-                    type=artifact.type.type, optional=artifact.optional, roles=roles, stages=artifact.stages
+                    type=artifact.type.type,
+                    optional=artifact.optional,
+                    roles=roles,
+                    stages=artifact.stages,
+                    description=artifact.desc,
                 )
             else:
                 raise ValueError(f"bad artifact: {artifact}")