pycityagent 2.0.0a71__cp312-cp312-macosx_11_0_arm64.whl → 2.0.0a73__cp312-cp312-macosx_11_0_arm64.whl

pycityagent/environment/sim/sim_env.py

@@ -51,6 +51,7 @@ class ControlSimEnv:
         log_dir: str,
         min_step_time: int = 1000,
         timeout: int = 5,
+        max_process: int = 32,
         sim_addr: Optional[str] = None,
     ):
         """
@@ -67,6 +68,7 @@ class ControlSimEnv:
         self._log_dir = log_dir
         self._min_step_time = min_step_time
         self._timeout = timeout
+        self._max_procs = max_process
 
         self._sim_config = _generate_yaml_config(map_file, start_step, total_step)
         # sim
@@ -99,6 +101,7 @@ class ControlSimEnv:
             assert self._sim_proc is None
             self.sim_port = find_free_port()
             config_base64 = encode_to_base64(self._sim_config)
+            os.environ["GOMAXPROCS"] = str(self._max_procs)
             self._sim_proc = Popen(
                 [
                     "pycityagent-sim",

{pycityagent-2.0.0a71.dist-info → pycityagent-2.0.0a73.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: pycityagent
-Version: 2.0.0a71
+Version: 2.0.0a73
 Summary: LLM-based city environment agent building library
 Author-email: Yuwei Yan <pinkgranite86@gmail.com>, Junbo Yan <yanjb20thu@gmali.com>, Jun Zhang <zhangjun990222@gmali.com>
 License: MIT License

{pycityagent-2.0.0a71.dist-info → pycityagent-2.0.0a73.dist-info}/RECORD

@@ -1,4 +1,10 @@
-pycityagent/pycityagent-sim,sha256=hVQtSEhdZ3lFxl5i6ubPuDSv31hke_W0V_6gP8CMxBs,35884466
+pycityagent-2.0.0a73.dist-info/RECORD,,
+pycityagent-2.0.0a73.dist-info/LICENSE,sha256=n2HPXiupinpyHMnIkbCf3OTYd3KMqbmldu1e7av0CAU,1084
+pycityagent-2.0.0a73.dist-info/WHEEL,sha256=VujM3ypTCyUW6hcTDdK2ej0ARVMxlU1Djlh_zWnDgqk,109
+pycityagent-2.0.0a73.dist-info/entry_points.txt,sha256=BZcne49AAIFv-hawxGnPbblea7X3MtAtoPyDX8L4OC4,132
+pycityagent-2.0.0a73.dist-info/top_level.txt,sha256=yOmeu6cSXmiUtScu53a3s0p7BGtLMaV0aff83EHCTic,43
+pycityagent-2.0.0a73.dist-info/METADATA,sha256=6YwxQbl14QnhjbHMwMPu4RoaBquNt8mSIXGJFaoD_CQ,9110
+pycityagent/pycityagent-sim,sha256=MYlRyvlpybVbiqKehzfV3GUbhatuQHNrxNWRBOs3nsU,35884466
 pycityagent/__init__.py,sha256=PUKWTXc-xdMG7px8oTNclodsILUgypANj2Z647sY63k,808
 pycityagent/pycityagent-ui,sha256=Ur95yZygIaZ5l_CDqP9394M5GQ66iV5PkcNPYFWqzvk,41225346
 pycityagent/metrics/mlflow_client.py,sha256=-iyh4BPVnBkluhmfucUzibCUnBX2iftkz4tVJJVxwHw,6958
@@ -58,7 +64,7 @@ pycityagent/environment/utils/const.py,sha256=1LqxnYJ8FSmq37fN5kIFlWLwycEDzFa8SF
 pycityagent/environment/sim/pause_service.py,sha256=UUTa4pTBOX_MGKki9QDgKId7scch0waAWF9xRVmNlAs,1701
 pycityagent/environment/sim/person_service.py,sha256=T1pRi5jDcivSInX7iGZoHKYZmgO7_t5MYtLhU12_LC4,10793
 pycityagent/environment/sim/aoi_service.py,sha256=bYVJDrOfDn8hgOORDU7PxMaT-fpqXA5Z0A080l2EPBc,1242
-pycityagent/environment/sim/sim_env.py,sha256=7HO5thoWMPQU_eZU2MolTLRgdS_x7I6F7qqdCWYxJJU,4496
+pycityagent/environment/sim/sim_env.py,sha256=aSdr3k3Scs2Mdd6PsBrABdz_7GKmV8kV4jo1Zw5N8nY,4625
 pycityagent/environment/sim/lane_service.py,sha256=bmf0zOWkxDzMEDlhTdHNRJRiTFGgzjbeqmex387gHZs,3964
 pycityagent/environment/sim/client.py,sha256=x4REWPRFG_C2s0-CWz8MnYalD_esfu2pPHf_vpsCG78,2852
 pycityagent/environment/sim/__init__.py,sha256=CyJc5Tey9MgzomZv0ynAcV0yaUlSE3bPbegYXar8wJc,601
@@ -89,9 +95,3 @@ pycityagent/cityagent/blocks/mobility_block.py,sha256=YmKhJW_srC6b6n_LvJujSO-eB-
 pycityagent/survey/models.py,sha256=g3xni4GcA1Py3vlGt6z4ltutjgQ4G0uINYAM8vKRJAw,5225
 pycityagent/survey/__init__.py,sha256=rxwou8U9KeFSP7rMzXtmtp2fVFZxK4Trzi-psx9LPIs,153
 pycityagent/survey/manager.py,sha256=tHkdeq4lTfAHwvgf4-udsXri0z2l6E00rEbvwl7SqRs,3439
-pycityagent-2.0.0a71.dist-info/RECORD,,
-pycityagent-2.0.0a71.dist-info/LICENSE,sha256=n2HPXiupinpyHMnIkbCf3OTYd3KMqbmldu1e7av0CAU,1084
-pycityagent-2.0.0a71.dist-info/WHEEL,sha256=VujM3ypTCyUW6hcTDdK2ej0ARVMxlU1Djlh_zWnDgqk,109
-pycityagent-2.0.0a71.dist-info/entry_points.txt,sha256=BZcne49AAIFv-hawxGnPbblea7X3MtAtoPyDX8L4OC4,132
-pycityagent-2.0.0a71.dist-info/top_level.txt,sha256=yOmeu6cSXmiUtScu53a3s0p7BGtLMaV0aff83EHCTic,43
-pycityagent-2.0.0a71.dist-info/METADATA,sha256=_fzsl3HAFmBCT9RR8lJYhRfxfFIanoNp30WEROFPbWU,9110