fix bug in engine.py

tsinghua-fib-lab · Jan 6, 2025 · 327e1d8 · 327e1d8
1 parent c581d2f
commit 327e1d8
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "python-moss"
-version = "1.1.2"
+version = "1.1.3"
 description = "MObility Simulation System"
 authors = [
   { name = "Jun Zhang", email = "[email protected]" },

diff --git a/python/src/moss/engine.py b/python/src/moss/engine.py
@@ -530,7 +530,7 @@ def get_finished_person_count(self) -> int:
         """
         persons = self.fetch_persons()
         status: NDArray[np.uint8] = persons["status"]
-        return (status == PersonStatus.FINISHED.value).sum()
+        return (status == FINISHED).sum()
 
     def get_finished_person_average_traveling_time(self) -> float:
         """
@@ -539,7 +539,7 @@ def get_finished_person_average_traveling_time(self) -> float:
         persons = self.fetch_persons()
         status: NDArray[np.uint8] = persons["status"]
         traveling_time = persons["traveling_time"]
-        return traveling_time[status == PersonStatus.FINISHED.value].mean()
+        return traveling_time[FINISHED].mean()
 
     def get_running_person_average_traveling_time(self) -> float:
         """
@@ -557,7 +557,7 @@ def get_departed_person_average_traveling_time(self) -> float:
         persons = self.fetch_persons()
         status: NDArray[np.uint8] = persons["status"]
         traveling_time = persons["traveling_time"]
-        return traveling_time[status != PersonStatus.SLEEP.value].mean()
+        return traveling_time[status != SLEEP].mean()
 
     def get_road_lane_plan_index(self, road_index: int) -> int:
         """