Multiclass-interface 1.18__tar.gz → 1.19__tar.gz

{multiclass_interface-1.18 → multiclass_interface-1.19/Multiclass_interface.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: Multiclass_interface
-Version: 1.18
+Version: 1.19
 Summary: Multiclass interface
 Author-email: "Mads M. Pedersen" <mmpe@dtu.dk>
 Project-URL: Homepage, https://gitlab.windenergy.dtu.dk/DYNAMIKS/multiclass_interface

{multiclass_interface-1.18 → multiclass_interface-1.19}/Multiclass_interface.egg-info/scm_version.json

@@ -1,7 +1,7 @@
 {
-  "tag": "1.18",
+  "tag": "1.19",
   "distance": 0,
-  "node": "gb934019843ac9640a5beae7880add16d3c7e3838",
+  "node": "g26b6778cd0dd2eedd49e1ca5a050e6c18834fc8d",
   "dirty": false,
   "branch": "HEAD",
   "node_date": "2026-10-23"

{multiclass_interface-1.18/Multiclass_interface.egg-info → multiclass_interface-1.19}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: Multiclass_interface
-Version: 1.18
+Version: 1.19
 Summary: Multiclass interface
 Author-email: "Mads M. Pedersen" <mmpe@dtu.dk>
 Project-URL: Homepage, https://gitlab.windenergy.dtu.dk/DYNAMIKS/multiclass_interface

{multiclass_interface-1.18 → multiclass_interface-1.19}/multiclass_interface/mpi_interface.py

@@ -258,7 +258,10 @@ def bcast_function_result(f):
     def wrapper(*args, **kwargs):
         if mpi:
             try:
-                v = f(*args, **kwargs)[0]
+                v = f(*args, **kwargs)
+                if not isinstance(v, list):
+                    v = [v]
+                v = v[0]
             except ChildProcessError:
                 v = None
 
@@ -267,7 +270,10 @@ def bcast_function_result(f):
             # broadcast the full list to all workers
             return comm.bcast(v, root=0)
         else:
-            return f(*args, **kwargs)
+            v = f(*args, **kwargs)
+            if not isinstance(v, list):
+                v = [v]
+            return v
     return wrapper
 
 

{multiclass_interface-1.18 → multiclass_interface-1.19}/multiclass_interface/tests/test_mpi.py

@@ -145,14 +145,40 @@ def test_non_collective_mpi2():
                 m.object.cls
 
 
-def test_non_collective_mpi2_no_mpi():
-    N = 3
+@pytest.mark.parametrize('N', [1, 3])
+def test_bcast_function_result_mci(N):
     with MultiClassInterface(MyTest, [(i + 10,) for i in range(N)]) as m:
 
         id_lst = mpi_interface.bcast_function_result(lambda: m.get_id())()
-        assert id_lst == [10, 11, 12], id_lst
+        assert id_lst == [10, 11, 12][:N], id_lst
+        id_lst = mpi_interface.bcast_function_result(lambda: m.return_input(['a', 'b', 'c'][:N]))()
+        assert id_lst == ["10 got: ('a',) and {}", "11 got: ('b',) and {}", "12 got: ('c',) and {}"][:N], id_lst
+
+        assert mpi_interface.bcast_function_result(lambda: 0)() == [0]
+
+
+@mpi_run_test
+def test_bcast_function_result_mpi1():
+    mpi_interface.activate_mpi(collective_mpi=False)
+    with MPIClassInterface(MyTest, [(i + 10,) for i in range(1)]) as m:
+        id_lst = mpi_interface.bcast_function_result(lambda: m.get_id())()
+        assert id_lst == [10, None, None, None], id_lst
+        id_lst = mpi_interface.bcast_function_result(lambda: m.return_input(['a']))()
+        assert id_lst == ["10 got: ('a',) and {}", None, None, None], id_lst
+
+        assert mpi_interface.bcast_function_result(lambda: mpi_interface.rank)() == [0, 1, 2, 3]
+
+
+@mpi_run_test
+def test_bcast_function_result_mpi3():
+    mpi_interface.activate_mpi(collective_mpi=False)
+    with MPIClassInterface(MyTest, [(i + 10,) for i in range(3)]) as m:
+        id_lst = mpi_interface.bcast_function_result(lambda: m.get_id())()
+        assert id_lst == [10, 11, 12, None], id_lst
         id_lst = mpi_interface.bcast_function_result(lambda: m.return_input(['a', 'b', 'c']))()
-        assert id_lst == ["10 got: ('a',) and {}", "11 got: ('b',) and {}", "12 got: ('c',) and {}"], id_lst
+        assert id_lst == ["10 got: ('a',) and {}", "11 got: ('b',) and {}", "12 got: ('c',) and {}", None], id_lst
+
+        assert mpi_interface.bcast_function_result(lambda: mpi_interface.rank)() == [0, 1, 2, 3]
 
 
 @mpi_run_test(4)

{multiclass_interface-1.18 → multiclass_interface-1.19}/multiclass_interface/version.py

@@ -18,7 +18,7 @@ version_tuple: tuple[int | str, ...]
 commit_id: str | None
 __commit_id__: str | None
 
-__version__ = version = '1.18'
-__version_tuple__ = version_tuple = (1, 18)
+__version__ = version = '1.19'
+__version_tuple__ = version_tuple = (1, 19)
 
-__commit_id__ = commit_id = 'gb93401984'
+__commit_id__ = commit_id = 'g26b6778cd'