langfun 0.1.2.dev202505010804__tar.gz → 0.1.2.dev202505020804__tar.gz

{langfun-0.1.2.dev202505010804 → langfun-0.1.2.dev202505020804}/PKG-INFO

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

{langfun-0.1.2.dev202505010804 → langfun-0.1.2.dev202505020804}/langfun/core/eval/v2/evaluation.py

@@ -167,7 +167,6 @@ class Evaluation(experiment_lib.Experiment):
       example.input = self.example_input_by_id(example.id)
 
     checkpointed = self._state.ckpt_example(example.id)
-    self._state.update(example, in_progress=True)
 
     with pg.timeit('evaluate') as timeit, lf.track_usages() as usage_summary:
       if checkpointed is None or checkpointed.has_error:
@@ -182,6 +181,7 @@ class Evaluation(experiment_lib.Experiment):
               f'contains error: {checkpointed.error}'
           )
         example.start_time = time.time()
+        self._state.update(example, in_progress=True)
         self._process(example, raise_if_has_error=raise_if_has_error)
       else:
         self.info(
@@ -189,6 +189,7 @@ class Evaluation(experiment_lib.Experiment):
             'is available and error free.'
         )
         example.start_time = checkpointed.start_time
+        self._state.update(example, in_progress=True)
 
         # Use the output and metadata obtained from the previous processing.
         example.output = checkpointed.output

{langfun-0.1.2.dev202505010804 → langfun-0.1.2.dev202505020804}/langfun/core/structured/querying.py

@@ -458,7 +458,9 @@ def query(
   if protocol is None:
     protocol = lf.context_value('__query_protocol__', 'python')
 
-  invocation_id = invocation_id or f'query@{uuid.uuid4().hex[-7:]}'
+  def _invocation_id():
+    return invocation_id or f'query@{uuid.uuid4().hex[-7:]}'
+
   # Multiple quries will be issued when `lm` is a list or `num_samples` is
   # greater than 1.
   if isinstance(lm, list) or num_samples != 1:
@@ -482,7 +484,7 @@ def query(
           protocol=protocol,
           returns_message=returns_message,
           skip_lm=skip_lm,
-          invocation_id=f'{invocation_id}:{i}',
+          invocation_id=f'{_invocation_id()}:{i}',
           **kwargs,
       )
     lm_list = lm if isinstance(lm, list) else [lm]
@@ -595,7 +597,7 @@ def query(
       metadata.pop('usage', None)
 
       invocation = QueryInvocation(
-          id=invocation_id,
+          id=_invocation_id(),
           input=pg.Ref(query_input),
           schema=(
               schema_lib.Schema.from_value(schema)

{langfun-0.1.2.dev202505010804 → langfun-0.1.2.dev202505020804}/langfun.egg-info/PKG-INFO

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