Immediately end sessions when all users are disconnected. (facebookre…

…search#2025)
SCAI-JHU · Aug 21, 2024 · 8f37d1d · 8f37d1d
1 parent 01da078
commit 8f37d1d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/examples/hitl/rearrange_v2/app_state_end_session.py b/examples/hitl/rearrange_v2/app_state_end_session.py
@@ -46,7 +46,8 @@ def __init__(
             self._status += f"\nError: {session.error}"
 
     def get_next_state(self) -> Optional[AppStateBase]:
-        if self._elapsed_time > SESSION_END_DELAY:
+        connected_user_count = len(self._app_data.connected_users)
+        if self._elapsed_time > SESSION_END_DELAY or connected_user_count == 0:
             self._end_session()
             return create_app_state_reset(self._app_service, self._app_data)
         return None