langfun 0.1.2.dev202509170804__py3-none-any.whl → 0.1.2.dev202509180804__py3-none-any.whl

langfun/env/__init__.py

@@ -35,3 +35,5 @@ from langfun.env.base_feature import BaseFeature
 
 from langfun.env import load_balancers
 from langfun.env.load_balancers import LoadBalancer
+
+# Google-internal imports.

langfun/env/base_feature.py

@@ -105,8 +105,9 @@ class BaseFeature(interface.Feature):
     self.__dict__.pop('name', None)
 
   @property
-  def sandbox(self) -> interface.Sandbox | None:
+  def sandbox(self) -> interface.Sandbox:
     """Returns the sandbox that the feature is running in."""
+    assert self._sandbox is not None, 'Feature has not been set up yet.'
     return self._sandbox
 
   #
@@ -123,6 +124,12 @@ class BaseFeature(interface.Feature):
 
   def teardown(self) -> None:
     """Tears down the feature."""
+    # If a sandbox is down during setting up, feature.shutdown might be called
+    # before the feature is setup. In this case, we don't need to teardown the
+    # feature.
+    if self._sandbox is None:
+      return
+
     interface.call_with_event(
         action=self._teardown,
         event_handler=self.on_teardown,

langfun/env/base_sandbox.py

@@ -91,14 +91,24 @@ class BaseSandbox(interface.Sandbox):
   def _start_session(self, session_id: str) -> None:
     """Starts a user session."""
     self._session_id = session_id
+    self._session_start_time = time.time()
+
     for feature in self._features.values():
       feature.setup_session(session_id)
 
   def _end_session(self) -> None:
-    """Ends the user session with the sandbox."""
-    for feature in self._features.values():
-      feature.teardown_session(self._session_id)
-    self._session_id = None
+    try:
+      for feature in self._features.values():
+        feature.teardown_session(self._session_id)
+    finally:
+      pg.logging.info(
+          '[%s]: User session %s ended. (lifetime: %.2f seconds).',
+          self.id,
+          self._session_id,
+          time.time() - self._session_start_time
+      )
+      self._session_id = None
+      self._session_start_time = None
 
   def _ping(self) -> None:
     """Implementation of ping for health checking."""
@@ -115,6 +125,7 @@ class BaseSandbox(interface.Sandbox):
         for name, feature in self.environment.features.items()
     })
     self._session_id = None
+    self._session_start_time = None
     self._alive = False
     self._start_time = None
 
@@ -197,6 +208,7 @@ class BaseSandbox(interface.Sandbox):
       return
 
     self._alive = False
+    shutdown_start_time = time.time()
     def shutdown_impl():
       self._teardown_features()
       self._shutdown()
@@ -205,8 +217,10 @@ class BaseSandbox(interface.Sandbox):
         self._housekeep_thread.join()
         self._housekeep_thread = None
       pg.logging.info(
-          '[%s]: Sandbox shutdown in %.2f seconds.',
-          self.id, time.time() - self._start_time
+          '[%s]: Sandbox shutdown in %.2f seconds. (lifetime: %.2f seconds)',
+          self.id,
+          time.time() - shutdown_start_time,
+          time.time() - self._start_time if self._start_time else 0
       )
 
     interface.call_with_event(

{langfun-0.1.2.dev202509170804.dist-info → langfun-0.1.2.dev202509180804.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langfun
-Version: 0.1.2.dev202509170804
+Version: 0.1.2.dev202509180804
 Summary: Langfun: Language as Functions.
 Home-page: https://github.com/google/langfun
 Author: Langfun Authors

{langfun-0.1.2.dev202509170804.dist-info → langfun-0.1.2.dev202509180804.dist-info}/RECORD

@@ -165,17 +165,17 @@ langfun/core/templates/demonstration.py,sha256=vCrgYubdZM5Umqcgp8NUVGXgr4P_c-fik
 langfun/core/templates/demonstration_test.py,sha256=SafcDQ0WgI7pw05EmPI2S4v1t3ABKzup8jReCljHeK4,2162
 langfun/core/templates/selfplay.py,sha256=yhgrJbiYwq47TgzThmHrDQTF4nDrTI09CWGhuQPNv-s,2273
 langfun/core/templates/selfplay_test.py,sha256=Ot__1P1M8oJfoTp-M9-PQ6HUXqZKyMwvZ5f7yQ3yfyM,2326
-langfun/env/__init__.py,sha256=zgiONH9pIpNTEL623jRmW2mMI-GhThzEegZoys_znrQ,1554
+langfun/env/__init__.py,sha256=YB44mvoCNiqWP48ITExQKZ3yHGoskFGB1pYRUEwWdbc,1582
 langfun/env/base_environment.py,sha256=BMdSpsk0uMCPklrZ7XQPpJ03IL3ZEzCIs3RlUfG8yNo,14215
-langfun/env/base_feature.py,sha256=SVvgoA68vafZZct5ZuHjPHblrCVv2Es4bLg5wgVTMww,4653
-langfun/env/base_sandbox.py,sha256=oSuUSd3zVgBxH14r9uX9SelyoXHzQQWzGHEiain8eKo,13690
+langfun/env/base_feature.py,sha256=C3s9ovV7MPOBGNM9evfABVkGxcfazX8A03Jb_RB1MW0,4936
+langfun/env/base_sandbox.py,sha256=SWDU3Mez7FVctpMwePbcavru3GkdTsC59neWz8MqTis,14126
 langfun/env/base_test.py,sha256=FuytEQJeKnxX_VEW0axdEoCg1A1d2rIVQss0qxoPPAQ,26739
 langfun/env/interface.py,sha256=4Q6ewOtj8XJyy8OEQqSYwogjzfvivFOljGS5txhlx_U,21534
 langfun/env/interface_test.py,sha256=nrqFLtpHd0vnOjoIX9cjV5WD1c-7J3WW29aogNu_94A,1395
 langfun/env/load_balancers.py,sha256=jbT3pB7QpPVovbXeJK1PNohUcrYrJNtWU8KaWzxO9nI,1963
 langfun/env/load_balancers_test.py,sha256=bAjC89zxmROpkl56BqvMU_yKOJ03Gtu8x8j56HO5HNc,4124
-langfun-0.1.2.dev202509170804.dist-info/licenses/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
-langfun-0.1.2.dev202509170804.dist-info/METADATA,sha256=Ps_Bme20ro1JnbT4nQj2Cs-euv4K-DpwcTg4BQr37u4,7380
-langfun-0.1.2.dev202509170804.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-langfun-0.1.2.dev202509170804.dist-info/top_level.txt,sha256=RhlEkHxs1qtzmmtWSwYoLVJAc1YrbPtxQ52uh8Z9VvY,8
-langfun-0.1.2.dev202509170804.dist-info/RECORD,,
+langfun-0.1.2.dev202509180804.dist-info/licenses/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
+langfun-0.1.2.dev202509180804.dist-info/METADATA,sha256=bdpN3vONnVYkei_3FGJQK7ohVxjle5cLnb3Gb_kgjyc,7380
+langfun-0.1.2.dev202509180804.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+langfun-0.1.2.dev202509180804.dist-info/top_level.txt,sha256=RhlEkHxs1qtzmmtWSwYoLVJAc1YrbPtxQ52uh8Z9VvY,8
+langfun-0.1.2.dev202509180804.dist-info/RECORD,,