fix: updated Workflow.get_topology() to use derived classes

ansys · Nov 21, 2024 · 849d50f · 849d50f
1 parent e75cfbc
commit 849d50f
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/src/ansys/dpf/core/dpf_operator.py b/src/ansys/dpf/core/dpf_operator.py
@@ -370,6 +370,7 @@ def _type_to_output_method(self):
             collection_base,
             any,
         )
+        from ansys.dpf.core.workflow_topology import workflow_topology
 
         out = [
             (any.Any, self._api.operator_getoutput_as_any),
@@ -466,6 +467,11 @@ def _type_to_output_method(self):
                 self._api.operator_getoutput_as_any,
                 lambda obj, type: any.Any(server=self._server, any_dpf=obj).cast(type),
             ),
+            (
+                workflow_topology.WorkflowTopology,
+                None,
+                "WorkflowTopology",
+            ),
         ]
         if hasattr(self._api, "operator_getoutput_generic_data_container"):
             out.append(

diff --git a/src/ansys/dpf/core/workflow.py b/src/ansys/dpf/core/workflow.py
@@ -959,7 +959,7 @@ def get_topology(self):
         workflow_to_workflow_topology_op.inputs.workflow.connect(self)
         workflow_topology_container = workflow_to_workflow_topology_op.outputs.workflow_topology()
 
-        return WorkflowTopology(workflow_topology_container)
+        return workflow_topology_container
 
     def __del__(self):
         try: