File tree Expand file tree Collapse file tree 2 files changed +3
-1
lines changed Expand file tree Collapse file tree 2 files changed +3
-1
lines changed Original file line number Diff line number Diff line change 77from ..utils import (
88 get_num_args ,
99 has_default_values ,
10- run_if_all_none ,
1110 mpi_scatter_from_rank0 ,
11+ run_if_all_none ,
1212)
1313
1414
Original file line number Diff line number Diff line change @@ -202,6 +202,7 @@ def get_num_args(func):
202202 params = inspect .signature (func ).parameters
203203 return len (params ) - ("self" in params )
204204
205+
205206def has_default_values (func ):
206207 """Check if the given function has default values for its parameters.
207208
@@ -214,6 +215,7 @@ def has_default_values(func):
214215 params = inspect .signature (func ).parameters .values ()
215216 return [param .default is not inspect .Parameter .empty for param in params ]
216217
218+
217219def mpi_scatter_from_rank0 (array , drop_last = True ):
218220 """Scatter the given array into continuous subarrays of equal size from rank 0 to all ranks.
219221
You can’t perform that action at this time.
0 commit comments