langfun 0.1.2.dev202509250804__py3-none-any.whl → 0.1.2.dev202509260805__py3-none-any.whl

langfun/env/test_utils.py

@@ -21,50 +21,47 @@ from langfun.env import base_environment
 from langfun.env import base_feature
 from langfun.env import base_sandbox
 from langfun.env import interface
+from langfun.env.event_handlers import base as event_handler_base
 import pyglove as pg
 
 
 class TestingEnvironment(base_environment.BaseEnvironment):
   """Testing environment for unit tests."""
 
+  housekeep_interval: float = 0.0
   simulate_start_error: Type[BaseException] | None = None
   simulate_shutdown_error: Type[BaseException] | None = None
   simulate_ping_error: Type[BaseException] | None = None
-  keepalive_interval: float | None = 60.0
   offline: bool = False
 
   @property
-  def id(self) -> interface.EnvironmentId:
-    return interface.EnvironmentId('testing-env')
+  def id(self) -> interface.Environment.Id:
+    return interface.Environment.Id('testing-env')
 
   def wait_for_housekeeping(self):
     housekeep_counter = self.housekeep_counter
     while self.housekeep_counter == housekeep_counter:
-      time.sleep(0.1)
+      time.sleep(0.01)
 
   def _create_sandbox(
       self,
       sandbox_id: str,
       reusable: bool,
       proactive_session_setup: bool,
+      keepalive_interval: float | None,
   ) -> base_sandbox.BaseSandbox:
-    if self.offline:
-      raise interface.EnvironmentError(
-          'Unknown environment error.',
-          environment=self,
-      )
     return TestingSandbox(
         environment=self,
-        id=interface.SandboxId(
+        id=interface.Sandbox.Id(
             environment_id=self.id,
             sandbox_id=sandbox_id
         ),
         reusable=reusable,
         proactive_session_setup=proactive_session_setup,
+        keepalive_interval=keepalive_interval,
         simulate_start_error=self.simulate_start_error,
         simulate_shutdown_error=self.simulate_shutdown_error,
         simulate_ping_error=self.simulate_ping_error,
-        keepalive_interval=self.keepalive_interval,
     )
 
 
@@ -88,15 +85,6 @@ class TestingSandbox(base_sandbox.BaseSandbox):
     else:
       raise error_type(message)
 
-  def wait_until(
-      self,
-      status: interface.Sandbox.Status | tuple[interface.Sandbox.Status, ...]
-  ) -> None:
-    if not isinstance(status, tuple):
-      status = (status,)
-    while self.status not in status:
-      time.sleep(0.1)
-
   def wait_until_not(
       self,
       status: interface.Sandbox.Status | tuple[interface.Sandbox.Status, ...]
@@ -104,12 +92,12 @@ class TestingSandbox(base_sandbox.BaseSandbox):
     if not isinstance(status, tuple):
       status = (status,)
     while self.status in status:
-      time.sleep(0.1)
+      time.sleep(0.01)
 
   def wait_until_next_housekeep(self) -> None:
     housekeep_counter = self.housekeep_counter
     while self.housekeep_counter == housekeep_counter:
-      time.sleep(0.1)
+      time.sleep(0.01)
 
   def _start(self) -> None:
     if self.simulate_start_error:
@@ -230,8 +218,8 @@ class TestingFeature(base_feature.BaseFeature):
       )
 
 
-class TestingEnvironmentEventHandler(
-    pg.Object, interface.EnvironmentEventHandler
+class TestingEventHandler(
+    pg.Object, event_handler_base.EventHandler
 ):
   """Testing environment event handler for unit tests."""
 
@@ -296,7 +284,6 @@ class TestingEnvironmentEventHandler(
       duration: float,
       error: BaseException | None
   ) -> None:
-    assert isinstance(environment, TestingEnvironment)
     assert duration > 0
     self._add_message(f'[{sandbox.id}] sandbox started', error)
 
@@ -459,7 +446,6 @@ class TestingEnvironmentEventHandler(
       **kwargs
   ) -> None:
     """Called when a sandbox activity is performed."""
-    assert duration > 0
     del environment, kwargs
     if session_id is None:
       log_id = sandbox.id

{langfun-0.1.2.dev202509250804.dist-info → langfun-0.1.2.dev202509260805.dist-info}/METADATA

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

{langfun-0.1.2.dev202509250804.dist-info → langfun-0.1.2.dev202509260805.dist-info}/RECORD

@@ -165,18 +165,22 @@ 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=1u5hRfO2i-shsPaoOsbYrnvwldo3Mvkd3BfqhV4dOC0,1538
-langfun/env/base_environment.py,sha256=FvAVfwfh3sT_XDrYH3adWo_ED6af-t6U3DuVamxv6YU,17636
-langfun/env/base_feature.py,sha256=UArKlAZMYpftakLD25apLw84RqtJA_XcA_nkFt_iBRQ,6162
-langfun/env/base_sandbox.py,sha256=Ena8GN_DaJj-4oMLH6kJQh1TMmBwo6_aU6IcdNWKNrw,36436
-langfun/env/base_test.py,sha256=sGNTylTxQsX9gfc8IMt6_7L9VgLn7q5W2xuDHl8LFuQ,57170
-langfun/env/interface.py,sha256=cuuBrFgCCEvKZx5EkKfIIKu1XfeijLr7sff9jyjrdg0,33359
-langfun/env/interface_test.py,sha256=nrqFLtpHd0vnOjoIX9cjV5WD1c-7J3WW29aogNu_94A,1395
+langfun/env/__init__.py,sha256=m6Y8y16ms9lytvO_r-Br8HmTp2rNDhb3R6JJaH5dEEk,1491
+langfun/env/base_environment.py,sha256=Vjpmck3k3UE3KaHq69dIuGbevbRJFlY6PvQEeZ18MtY,18087
+langfun/env/base_feature.py,sha256=dztFq1lcdXgUh3m8pNRC3vruyIdHCob27reiXz_02q8,6436
+langfun/env/base_sandbox.py,sha256=tR__USJZojDrGtnhmDqkbCHrvcrNUDrTwc9mpStyYZ0,36721
+langfun/env/base_test.py,sha256=cBZEcoJG16L_XLVyTgSCxUxy1cukrkpSNNEgiM5LZb4,59137
+langfun/env/interface.py,sha256=2Amf-_op7dGRF8c4-wYxcFxs1UCBYz1AB20Lk7__V4E,25724
+langfun/env/interface_test.py,sha256=hfQn4RRTEo1YfVHXTPzH1puzD14BTo8R_5v1IpXVZ90,1398
 langfun/env/load_balancers.py,sha256=qRhCthqzjZIQBwta8qC1C0s0J-VQAVomJQqI7Nqv-r4,1948
-langfun/env/load_balancers_test.py,sha256=gJKJ3K5_WS7hsz9Jv3nPEjlb04Z6MmrZpfCu86siYKU,3967
-langfun/env/test_utils.py,sha256=piNl09bQUW0GzEyQkyTCCJwzfxQ3z8s3M1lYcHCnmdo,14248
-langfun-0.1.2.dev202509250804.dist-info/licenses/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
-langfun-0.1.2.dev202509250804.dist-info/METADATA,sha256=pAaeHsBcf9oirhLuF2RMulQm-r08T0lz48kO4xWv6Rw,7380
-langfun-0.1.2.dev202509250804.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-langfun-0.1.2.dev202509250804.dist-info/top_level.txt,sha256=RhlEkHxs1qtzmmtWSwYoLVJAc1YrbPtxQ52uh8Z9VvY,8
-langfun-0.1.2.dev202509250804.dist-info/RECORD,,
+langfun/env/load_balancers_test.py,sha256=KIDIwZeQHcW0fCL5ScZD0lm0QV4_X2y-rMqdd6R1gLc,3737
+langfun/env/test_utils.py,sha256=Cx4J69r_dLcmDyzEVqOzAZ1Gtk5CnbM2K6nSnRHzIAc,13874
+langfun/env/event_handlers/__init__.py,sha256=sZTUeYpnUMzfH6AghhT-hnBHnFxg5UaH-AOmw3YbyMM,376
+langfun/env/event_handlers/base.py,sha256=w2pdAKfGL-TDTQD2lQ-c_BD3jWeAXF5SYBkAj2GNRz0,7532
+langfun/env/event_handlers/event_logger.py,sha256=4jDV6siGRXR95QFJj_9rnQ_BYMaR6IaPb-wuZ7YzCls,11200
+langfun/env/event_handlers/event_logger_test.py,sha256=OpObIB9IXuuNaRFxOuYcmV4RV6sUTbH5zr3-R7B5UGQ,8536
+langfun-0.1.2.dev202509260805.dist-info/licenses/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
+langfun-0.1.2.dev202509260805.dist-info/METADATA,sha256=Tee4run8k20LEd7W2ZXxM6E_TzH4g6GFQF-ude-fhS4,7380
+langfun-0.1.2.dev202509260805.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+langfun-0.1.2.dev202509260805.dist-info/top_level.txt,sha256=RhlEkHxs1qtzmmtWSwYoLVJAc1YrbPtxQ52uh8Z9VvY,8
+langfun-0.1.2.dev202509260805.dist-info/RECORD,,