Merge pull request #5314 from FederatedAI/feature-2.0.0-rc1-flow

Feature 2.0.0 rc1 flow
FederatedAI · Dec 11, 2023 · 95a29c1 · 95a29c1
2 parents 711d69b + 71ce84a
commit 95a29c1
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/python/fate/components/components/dataframe_transformer.py b/python/fate/components/components/dataframe_transformer.py
@@ -13,15 +13,15 @@
 #  See the License for the specific language governing permissions and
 #  limitations under the License.
 #
-from fate.components.core import LOCAL, Role, cpn
+from fate.components.core import LOCAL, Role, cpn, GUEST, HOST
 
 
-@cpn.component(roles=[LOCAL])
+@cpn.component(roles=[LOCAL, GUEST, HOST])
 def dataframe_transformer(
     ctx,
     role: Role,
-    table: cpn.table_input(roles=[LOCAL]),
-    dataframe_output: cpn.dataframe_output(roles=[LOCAL]),
+    table: cpn.table_input(roles=[LOCAL, GUEST, HOST]),
+    dataframe_output: cpn.dataframe_output(roles=[LOCAL, GUEST, HOST]),
     namespace: cpn.parameter(type=str, default=None, optional=True),
     name: cpn.parameter(type=str, default=None, optional=True),
     site_name: cpn.parameter(type=str, default=None, optional=True),