Update the exception

ansys · Sep 7, 2023 · 4e2c7af · 4e2c7af
1 parent ada607e
commit 4e2c7af
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/ansys/dpf/core/data_sources.py b/src/ansys/dpf/core/data_sources.py
@@ -78,7 +78,7 @@ def __init__(self, result_path=None, data_sources=None, server=None):
                 self._internal_obj = core_api.data_processing_duplicate_object_reference(
                     data_sources
                 )
-            elif isinstance(data_sources, data_sources_pb2.DataSources):
+            elif isinstance(data_sources, data_sources_pb2.DataSources) or isinstance(data_sources, int):
                 # It should be a message (usually from a call to operator_getoutput_data_sources)
                 self._internal_obj = data_sources
             else: