aspire-inference 0.1.0a11__tar.gz → 0.1.0a12__tar.gz

{aspire_inference-0.1.0a11 → aspire_inference-0.1.0a12}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aspire-inference
-Version: 0.1.0a11
+Version: 0.1.0a12
 Summary: Accelerate Sequential Posterior Inference via REuse
 Author-email: "Michael J. Williams" <michaeljw1@googlemail.com>
 License: MIT

{aspire_inference-0.1.0a11 → aspire_inference-0.1.0a12}/aspire_inference.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aspire-inference
-Version: 0.1.0a11
+Version: 0.1.0a12
 Summary: Accelerate Sequential Posterior Inference via REuse
 Author-email: "Michael J. Williams" <michaeljw1@googlemail.com>
 License: MIT

{aspire_inference-0.1.0a11 → aspire_inference-0.1.0a12}/src/aspire/aspire.py

@@ -20,6 +20,7 @@ from .transforms import (
 )
 from .utils import (
     AspireFile,
+    function_id,
     load_from_h5_file,
     recursively_save_to_h5_file,
     resolve_xp,
@@ -687,8 +688,8 @@ class Aspire:
             method of the sampler.
         """
         config = {
-            "log_likelihood": self.log_likelihood.__name__,
-            "log_prior": self.log_prior.__name__,
+            "log_likelihood": function_id(self.log_likelihood),
+            "log_prior": function_id(self.log_prior),
             "dims": self.dims,
             "parameters": self.parameters,
             "periodic_parameters": self.periodic_parameters,

{aspire_inference-0.1.0a11 → aspire_inference-0.1.0a12}/src/aspire/utils.py

@@ -1,5 +1,6 @@
 from __future__ import annotations
 
+import functools
 import inspect
 import logging
 import pickle
@@ -911,3 +912,23 @@ def track_calls(wrapped=None):
         return wrapped_func(*args, **kwargs)
 
     return wrapper(wrapped) if wrapped else wrapper
+
+
+def function_id(fn: Any) -> str:
+    """Get a unique identifier for a function.
+
+    Parameters
+    ----------
+    fn : Any
+        The function to get the identifier for.
+
+    Returns
+    -------
+    str
+        The unique identifier for the function.
+    """
+    if isinstance(fn, functools.partial):
+        base = fn.func
+    else:
+        base = fn
+    return f"{base.__module__}:{getattr(base, '__qualname__', type(base).__name__)}"

{aspire_inference-0.1.0a11 → aspire_inference-0.1.0a12}/tests/test_utils.py

@@ -1,3 +1,4 @@
+import functools
 import pickle
 
 import array_api_compat.numpy as np_xp
@@ -6,7 +7,7 @@ import h5py
 import jax.numpy as jnp
 import pytest
 
-from aspire.utils import convert_dtype, dump_state, resolve_dtype
+from aspire.utils import convert_dtype, dump_state, function_id, resolve_dtype
 
 
 def _dtype_name(dtype):
@@ -85,3 +86,18 @@ def test_dump_state_round_trip(tmp_path):
         stored = fp["checkpoints"]["state"][...]
     restored = pickle.loads(stored.tobytes())
     assert restored == state
+
+
+# Define a simple function for testing
+def _foo(x):
+    return x
+
+
+@pytest.mark.parametrize(
+    "fn", [lambda x: x, functools.partial(lambda x, y: x, y=1), _foo]
+)
+def test_function_id(fn):
+    fn_id = function_id(fn)
+    assert isinstance(fn_id, str)
+    # Calling again should give the same result
+    assert fn_id == function_id(fn)