mplang-nightly 0.1.dev173__tar.gz → 0.1.dev175__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (174) hide show
  1. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/PKG-INFO +1 -1
  2. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/examples/stax_nn/stax_nn.py +4 -5
  3. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/examples/xgboost/sgb.py +187 -153
  4. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/examples/xgboost/sgb_test.py +36 -38
  5. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/__init__.py +78 -1
  6. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/analysis/diagram.py +2 -2
  7. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/core/__init__.py +4 -0
  8. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/core/mpir.py +4 -4
  9. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/core/primitive.py +25 -11
  10. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/device.py +28 -30
  11. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/runtime/driver.py +2 -2
  12. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/runtime/server.py +2 -2
  13. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/runtime/simulation.py +3 -3
  14. mplang_nightly-0.1.dev175/mplang/simp/api.py +96 -0
  15. mplang_nightly-0.1.dev173/mplang/simp/__init__.py → mplang_nightly-0.1.dev175/mplang/simp/party.py +16 -142
  16. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/simp/random.py +8 -8
  17. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/pyproject.toml +5 -1
  18. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tests/analysis/test_diagram.py +14 -17
  19. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tests/core/test_mpir.py +48 -48
  20. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tests/core/test_primitive.py +4 -5
  21. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tests/integration/test_crypto_roundtrip.py +37 -39
  22. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tests/integration/test_http_e2e.py +56 -66
  23. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tests/integration/test_symbols_roundtrip.py +21 -28
  24. mplang_nightly-0.1.dev175/tests/integration/test_tee_workflow.py +83 -0
  25. mplang_nightly-0.1.dev175/tests/integration/test_tutorials.py +318 -0
  26. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tests/integration/test_unused_param_integration.py +45 -46
  27. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tests/kernels/test_sql_duckdb.py +6 -6
  28. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tests/ops/test_basic_pack.py +7 -7
  29. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tests/ops/test_sql.py +2 -2
  30. mplang_nightly-0.1.dev175/tests/runtime/__init__.py +13 -0
  31. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tests/runtime/test_simulation.py +123 -77
  32. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tests/simp/test_mpi.py +48 -22
  33. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tests/simp/test_smpc.py +9 -8
  34. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tests/simp/test_sugar.py +13 -13
  35. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tutorials/0_basic.py +52 -53
  36. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tutorials/10_analysis.py +5 -5
  37. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tutorials/1_condition.py +46 -40
  38. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tutorials/2_whileloop.py +36 -37
  39. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tutorials/3_device.py +10 -10
  40. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tutorials/4_simulation.py +18 -19
  41. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tutorials/5_ir_dump.py +11 -12
  42. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tutorials/6_advanced.py +11 -12
  43. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tutorials/7_stdio.py +19 -21
  44. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tutorials/8_phe.py +15 -16
  45. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tutorials/9_tee.py +19 -20
  46. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tutorials/pitfalls/late_binding.py +16 -17
  47. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tutorials/pitfalls/rand.py +23 -24
  48. mplang_nightly-0.1.dev173/tests/integration/test_tutorials.py +0 -320
  49. mplang_nightly-0.1.dev173/tests/ops/test_crypto_tee.py +0 -84
  50. mplang_nightly-0.1.dev173/tests/simp/test_simp.py +0 -52
  51. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/.gitignore +0 -0
  52. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/LICENSE +0 -0
  53. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/README.md +0 -0
  54. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/examples/conf/3pc.yaml +0 -0
  55. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/examples/stax_nn/README.md +0 -0
  56. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/examples/stax_nn/models.py +0 -0
  57. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/examples/xgboost/hist_jax.py +0 -0
  58. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/examples/xgboost/hist_jax_test.py +0 -0
  59. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/examples/xgboost/naive_np.py +0 -0
  60. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/examples/xgboost/readme.md +0 -0
  61. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/hatch_build.py +0 -0
  62. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/analysis/__init__.py +0 -0
  63. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/core/cluster.py +0 -0
  64. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/core/comm.py +0 -0
  65. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/core/context_mgr.py +0 -0
  66. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/core/dtype.py +0 -0
  67. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/core/expr/__init__.py +0 -0
  68. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/core/expr/ast.py +0 -0
  69. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/core/expr/evaluator.py +0 -0
  70. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/core/expr/printer.py +0 -0
  71. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/core/expr/transformer.py +0 -0
  72. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/core/expr/utils.py +0 -0
  73. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/core/expr/visitor.py +0 -0
  74. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/core/expr/walk.py +0 -0
  75. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/core/interp.py +0 -0
  76. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/core/mask.py +0 -0
  77. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/core/mpobject.py +0 -0
  78. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/core/mptype.py +0 -0
  79. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/core/pfunc.py +0 -0
  80. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/core/table.py +0 -0
  81. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/core/tensor.py +0 -0
  82. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/core/tracer.py +0 -0
  83. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/host.py +0 -0
  84. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/kernels/__init__.py +0 -0
  85. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/kernels/base.py +0 -0
  86. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/kernels/basic.py +0 -0
  87. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/kernels/context.py +0 -0
  88. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/kernels/crypto.py +0 -0
  89. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/kernels/mock_tee.py +0 -0
  90. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/kernels/phe.py +0 -0
  91. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/kernels/spu.py +0 -0
  92. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/kernels/sql_duckdb.py +0 -0
  93. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/kernels/stablehlo.py +0 -0
  94. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/kernels/value.py +0 -0
  95. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/ops/__init__.py +0 -0
  96. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/ops/base.py +0 -0
  97. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/ops/basic.py +0 -0
  98. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/ops/crypto.py +0 -0
  99. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/ops/ibis_cc.py +0 -0
  100. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/ops/jax_cc.py +0 -0
  101. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/ops/phe.py +0 -0
  102. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/ops/spu.py +0 -0
  103. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/ops/sql_cc.py +0 -0
  104. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/ops/tee.py +0 -0
  105. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/protos/v1alpha1/mpir_pb2.py +0 -0
  106. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/protos/v1alpha1/mpir_pb2.pyi +0 -0
  107. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/protos/v1alpha1/value_pb2.py +0 -0
  108. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/protos/v1alpha1/value_pb2.pyi +0 -0
  109. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/runtime/__init__.py +0 -0
  110. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/runtime/cli.py +0 -0
  111. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/runtime/client.py +0 -0
  112. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/runtime/communicator.py +0 -0
  113. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/runtime/data_providers.py +0 -0
  114. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/runtime/exceptions.py +0 -0
  115. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/runtime/http_api.md +0 -0
  116. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/runtime/link_comm.py +0 -0
  117. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/runtime/session.py +0 -0
  118. {mplang_nightly-0.1.dev173/mplang/utils → mplang_nightly-0.1.dev175/mplang/simp}/__init__.py +0 -0
  119. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/simp/mpi.py +0 -0
  120. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/simp/smpc.py +0 -0
  121. {mplang_nightly-0.1.dev173/tests/device → mplang_nightly-0.1.dev175/mplang/utils}/__init__.py +0 -0
  122. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/utils/crypto.py +0 -0
  123. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/utils/func_utils.py +0 -0
  124. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/utils/spu_utils.py +0 -0
  125. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/mplang/utils/table_utils.py +0 -0
  126. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tests/__init__.py +0 -0
  127. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tests/conftest.py +0 -0
  128. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tests/core/__init__.py +0 -0
  129. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tests/core/expr/__init__.py +0 -0
  130. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tests/core/expr/conftest.py +0 -0
  131. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tests/core/expr/test_ast.py +0 -0
  132. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tests/core/expr/test_printer.py +0 -0
  133. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tests/core/expr/test_utils.py +0 -0
  134. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tests/core/expr/test_walk.py +0 -0
  135. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tests/core/test_cluster.py +0 -0
  136. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tests/core/test_dtype.py +0 -0
  137. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tests/core/test_mask.py +0 -0
  138. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tests/core/test_mptype.py +0 -0
  139. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tests/core/test_table.py +0 -0
  140. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tests/core/test_tensor.py +0 -0
  141. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tests/core/test_tracer.py +0 -0
  142. {mplang_nightly-0.1.dev173/tests/ops → mplang_nightly-0.1.dev175/tests/device}/__init__.py +0 -0
  143. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tests/device/test_device_basic.py +0 -0
  144. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tests/integration/README.md +0 -0
  145. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tests/kernels/test_basic.py +0 -0
  146. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tests/kernels/test_debug_print.py +0 -0
  147. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tests/kernels/test_kernel_binding.py +0 -0
  148. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tests/kernels/test_phe.py +0 -0
  149. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tests/kernels/test_spu.py +0 -0
  150. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tests/kernels/test_stablehlo.py +0 -0
  151. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tests/kernels/test_value.py +0 -0
  152. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tests/kernels/test_value_serde.py +0 -0
  153. {mplang_nightly-0.1.dev173/tests/runtime → mplang_nightly-0.1.dev175/tests/ops}/__init__.py +0 -0
  154. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tests/ops/dummy.py +0 -0
  155. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tests/ops/test_feop_base.py +0 -0
  156. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tests/ops/test_ibis.py +0 -0
  157. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tests/ops/test_ibis_cc.py +0 -0
  158. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tests/ops/test_jax_cc.py +0 -0
  159. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tests/ops/test_phe.py +0 -0
  160. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tests/ops/test_spu.py +0 -0
  161. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tests/ops/test_spu_defensive.py +0 -0
  162. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tests/ops/test_table_tensor_conversion.py +0 -0
  163. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tests/runtime/test_cli.py +0 -0
  164. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tests/runtime/test_communicator.py +0 -0
  165. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tests/runtime/test_driver.py +0 -0
  166. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tests/runtime/test_server.py +0 -0
  167. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tests/simp/test_random.py +0 -0
  168. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tests/utils/__init__.py +0 -0
  169. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tests/utils/server_fixtures.py +0 -0
  170. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tests/utils/test_func_utils.py +0 -0
  171. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tests/utils/test_spu_utils.py +0 -0
  172. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tests/utils/test_table_utils.py +0 -0
  173. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tutorials/__init__.py +0 -0
  174. {mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/tutorials/run.sh +0 -0
{mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: mplang-nightly
3
- Version: 0.1.dev173
3
+ Version: 0.1.dev175
4
4
  Summary: Multi-Party Programming Language
5
5
  Author-email: SecretFlow Team <secretflow-contact@service.alipay.com>
6
6
  License: Apache License
{mplang_nightly-0.1.dev173 → mplang_nightly-0.1.dev175}/examples/stax_nn/stax_nn.py RENAMED
@@ -27,9 +27,8 @@ import yaml
27
27
  from jax.example_libraries import stax
28
28
  from sklearn.metrics import accuracy_score
29
29
 
30
- import mplang
30
+ import mplang as mp
31
31
  import mplang.device as mpd
32
- from mplang.core.cluster import ClusterSpec
33
32
 
34
33
  parser = argparse.ArgumentParser(description="distributed driver.")
35
34
  parser.add_argument("--model", default="network_a", type=str)
@@ -239,9 +238,9 @@ def main():
239
238
 
240
239
  with open(args.config) as file:
241
240
  conf = yaml.safe_load(file)
242
- cluster_spec = ClusterSpec.from_dict(conf)
243
- driver = mplang.Driver(cluster_spec, timeout=600)
244
- mplang.set_ctx(driver)
241
+ cluster_spec = mp.ClusterSpec.from_dict(conf)
242
+ driver = mp.Driver(cluster_spec, timeout=600)
243
+ mp.set_ctx(driver)
245
244
 
246
245
  # with open(args.config) as file:
247
246
  # conf = json.load(file)