mplang-nightly 0.1.dev253__py3-none-any.whl → 0.1.dev254__py3-none-any.whl

mplang/v2/backends/simp_driver/mem.py

@@ -154,6 +154,10 @@ class SimpMemDriver(SimpDriver):
         self._workers = workers
         self._mesh = mesh
 
+    def shutdown(self) -> None:
+        """Shutdown the local memory driver and its mesh."""
+        self._mesh.shutdown()
+
     @property
     def world_size(self) -> int:
         return self._world_size

mplang/v2/cli.py

@@ -62,7 +62,7 @@ def run_worker(
     signal.signal(signal.SIGINT, signal.SIG_DFL)
     signal.signal(signal.SIGTERM, signal.SIG_DFL)
 
-    from mplang.v2.backends.simp_http_worker import create_worker_app
+    from mplang.v2.backends.simp_worker.http import create_worker_app
 
     app = create_worker_app(rank, world_size, endpoints, spu_endpoints)
 

mplang/v2/edsl/context.py

@@ -69,7 +69,7 @@ class Context(ABC):
     """
 
     def __init__(self) -> None:
-        self._state: dict[str, Any] = {}
+        self._states: dict[str, Any] = {}
 
     # =========================================================================
     # State Management
@@ -82,7 +82,7 @@ class Context(ABC):
             key: State key (e.g., "dialect.simp", "device.cluster")
             value: State value
         """
-        self._state[key] = value
+        self._states[key] = value
 
     def get_state(self, key: str, default: Any = None) -> Any:
         """Get attached state by key.
@@ -94,7 +94,7 @@ class Context(ABC):
         Returns:
             State value or default
         """
-        return self._state.get(key, default)
+        return self._states.get(key, default)
 
     def has_state(self, key: str) -> bool:
         """Check if state exists.
@@ -105,7 +105,7 @@ class Context(ABC):
         Returns:
             True if state exists
         """
-        return key in self._state
+        return key in self._states
 
     # =========================================================================
     # Abstract Methods

mplang/v2/libs/ml/__init__.py

@@ -0,0 +1,23 @@
+# Copyright 2025 Ant Group Co., Ltd.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""Machine Learning algorithms for secure multi-party computation."""
+
+from mplang.v2.libs.ml.sgb import SecureBoost, Tree, TreeEnsemble
+
+__all__ = [
+    "SecureBoost",
+    "Tree",
+    "TreeEnsemble",
+]