gpflow 2.9.2__tar.gz → 2.10.0__tar.gz

{gpflow-2.9.2/gpflow.egg-info → gpflow-2.10.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: gpflow
-Version: 2.9.2
+Version: 2.10.0
 Summary: Gaussian process methods in TensorFlow
 Home-page: https://www.gpflow.org
 Author: James Hensman, Alex Matthews
@@ -9,7 +9,6 @@ License: Apache License 2.0
 Project-URL: Source on GitHub, https://github.com/GPflow/GPflow
 Project-URL: Documentation, https://gpflow.github.io/GPflow/
 Keywords: machine-learning gaussian-processes kernels tensorflow
-Platform: UNKNOWN
 Classifier: Development Status :: 5 - Production/Stable
 Classifier: Environment :: GPU :: NVIDIA CUDA
 Classifier: Intended Audience :: Developers
@@ -217,4 +216,3 @@ which can be cited by users.
   url = {https://arxiv.org/abs/2003.01115}
 }
 ```
-

{gpflow-2.9.2 → gpflow-2.10.0}/RELEASE.md

@@ -33,7 +33,7 @@ This release contains contributions from:
 <INSERT>, <NAME>, <HERE>, <USING>, <GITHUB>, <HANDLE>
 
 
-# Release 2.10.0 (next release)
+# Release 2.11.0 (next release)
 
 <INSERT SMALL BLURB ABOUT RELEASE FOCUS AREA AND POTENTIAL TOOLCHAIN CHANGES>
 
@@ -65,6 +65,19 @@ This release contains contributions from:
 
 <INSERT>, <NAME>, <HERE>, <USING>, <GITHUB>, <HANDLE>
 
+# Release 2.10.0
+
+This release adds a utility function for converting tf.Modules to float32.
+
+## Improvements
+
+* Add the `freeze_as_float32` utility function, which returns a frozen deepcopy of a module with all values converted to tf.float32.
+
+## Thanks to our Contributors
+
+This release contains contributions from:
+
+uri-granta
 
 # Release 2.9.2
 

gpflow-2.10.0/VERSION

@@ -0,0 +1 @@
+2.10.0

{gpflow-2.9.2 → gpflow-2.10.0}/doc/sphinx/conf.py

@@ -24,7 +24,7 @@ copyright = "2023, The GPflow Contributors"
 author = "The GPflow Contributors"
 
 # The full version, including alpha/beta/rc tags
-release = "2.9.2"
+release = "2.10.0"
 
 # -- General configuration ---------------------------------------------------
 

{gpflow-2.9.2 → gpflow-2.10.0}/gpflow/keras.py

@@ -23,8 +23,11 @@
 # >> from tf_keras.optimizers import Adam
 
 try:
+    # check whether a (recent) tf_keras is present
     import tf_keras
-except ModuleNotFoundError:
+
+    tf_keras.optimizers
+except (ModuleNotFoundError, AttributeError):  # pragma: no cover
     import tensorflow.keras as tf_keras
 
 tf_keras = tf_keras

{gpflow-2.9.2 → gpflow-2.10.0}/gpflow/optimizers/mcmc.py

@@ -73,9 +73,7 @@ class SamplingHelper:
         return self._variables
 
     @property
-    def target_log_prob_fn(
-        self,
-    ) -> Callable[..., Tuple[tf.Tensor, Callable[..., Tuple[tf.Tensor, Sequence[None]]]]]:
+    def target_log_prob_fn(self) -> Callable[..., tf.Tensor]:
         """
         The target log probability, adjusted to allow for optimisation to occur on the tracked
         unconstrained underlying variables.

{gpflow-2.9.2 → gpflow-2.10.0}/gpflow/posteriors.py

@@ -967,10 +967,13 @@ class FullyCorrelatedPosterior(BasePosterior):
             cov = tf.linalg.adjoint(cov)
 
         mean = tf.reshape(mean, (N, K))
-        if full_cov == full_output_cov:
-            cov_shape = (N, K, N, K) if full_cov else (N, K)
-        else:
-            cov_shape = (K, N, N) if full_cov else (N, K, K)
+        # the nested if-then-else is to make both mypy and the tensorflow
+        # compiler happy across multiple different versions!
+        cov_shape: Tuple[int, ...] = (
+            ((N, K, N, K) if full_cov else (N, K))
+            if full_cov == full_output_cov
+            else ((K, N, N) if full_cov else (N, K, K))
+        )
         cov = tf.reshape(cov, cov_shape)
 
         return mean, cov

{gpflow-2.9.2 → gpflow-2.10.0}/gpflow/utilities/__init__.py

@@ -5,6 +5,7 @@ from .multipledispatch import Dispatcher
 from .traversal import (
     deepcopy,
     freeze,
+    freeze_as_float32,
     leaf_components,
     multiple_assign,
     parameter_dict,
@@ -24,6 +25,7 @@ __all__ = [
     "bijectors",
     "deepcopy",
     "freeze",
+    "freeze_as_float32",
     "is_variable",
     "leaf_components",
     "misc",

{gpflow-2.9.2 → gpflow-2.10.0}/gpflow/utilities/traversal.py

@@ -30,6 +30,7 @@ from ..config import default_summary_fmt
 __all__ = [
     "deepcopy",
     "freeze",
+    "freeze_as_float32",
     "leaf_components",
     "multiple_assign",
     "parameter_dict",
@@ -98,10 +99,10 @@ def print_summary(module: tf.Module, fmt: Optional[str] = None) -> None:
     """
     fmt = fmt if fmt is not None else default_summary_fmt()
     if fmt == "notebook":
-        from IPython.core.display import HTML, display
+        from IPython.core.display import HTML, display_html
 
         tab = tabulate_module_summary(module, "html")
-        display(HTML(tab))
+        display_html(HTML(tab))
     else:
         print(tabulate_module_summary(module, fmt))
 
@@ -277,6 +278,21 @@ def freeze(input_module: M) -> M:
     return module_copy
 
 
+def freeze_as_float32(input_module: M) -> M:
+    """
+    Returns a frozen deepcopy of the input tf.Module with all values converted to tf.float32.
+
+    :param input_module: tf.Module or gpflow.Module.
+    :return: Returns a float32 frozen deepcopy of an input object.
+    """
+    objects_to_freeze = _get_leaf_components(input_module)
+    memo_tensors = {
+        id(v): tf.cast(tf.convert_to_tensor(v), dtype=tf.float32)
+        for v in objects_to_freeze.values()
+    }
+    return deepcopy(input_module, memo_tensors)
+
+
 def traverse_module(
     m: TraverseInput,
     acc: Accumulator[State],

{gpflow-2.9.2 → gpflow-2.10.0/gpflow.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: gpflow
-Version: 2.9.2
+Version: 2.10.0
 Summary: Gaussian process methods in TensorFlow
 Home-page: https://www.gpflow.org
 Author: James Hensman, Alex Matthews
@@ -9,7 +9,6 @@ License: Apache License 2.0
 Project-URL: Source on GitHub, https://github.com/GPflow/GPflow
 Project-URL: Documentation, https://gpflow.github.io/GPflow/
 Keywords: machine-learning gaussian-processes kernels tensorflow
-Platform: UNKNOWN
 Classifier: Development Status :: 5 - Production/Stable
 Classifier: Environment :: GPU :: NVIDIA CUDA
 Classifier: Intended Audience :: Developers
@@ -217,4 +216,3 @@ which can be cited by users.
   url = {https://arxiv.org/abs/2003.01115}
 }
 ```
-

{gpflow-2.9.2 → gpflow-2.10.0}/gpflow.egg-info/requires.txt

@@ -1,7 +1,7 @@
 check_shapes>=1.0.0
 deprecated
 multipledispatch>=0.6
-numpy
+numpy<2
 packaging
 scipy
 setuptools>=41.0.0

{gpflow-2.9.2 → gpflow-2.10.0}/setup.py

@@ -11,7 +11,7 @@ requirements = [
     "check_shapes>=1.0.0",
     "deprecated",
     "multipledispatch>=0.6",
-    "numpy",
+    "numpy<2",
     "packaging",
     "scipy",
     "setuptools>=41.0.0",  # to satisfy dependency constraints

{gpflow-2.9.2 → gpflow-2.10.0}/tests/gpflow/test_initial_value.py

@@ -74,7 +74,7 @@ def test_assignment_value(path: Path) -> None:
             these problems.
             """
 
-            self.bad_assigns: List[ast.AST] = []
+            self.bad_assigns: List[Union[ast.Assign, ast.AugAssign, ast.AnnAssign]] = []
             """
             List of bad assignments, so that we can report multiple errors at the same time.
             """

{gpflow-2.9.2 → gpflow-2.10.0}/tests/gpflow/utilities/test_deepcopy.py

@@ -14,7 +14,7 @@ from gpflow.base import TensorType
 
 class A(tf.Module):
     def __init__(self) -> None:
-        self.var = tf.Variable([1.0])
+        self.var = tf.Variable([1.0], dtype=tf.float64)
         shift = tf.Variable(1e-6)
         self.bijector = tfp.bijectors.Chain([tfp.bijectors.Softplus(), tfp.bijectors.Shift(shift)])
 
@@ -87,6 +87,21 @@ def test_freeze() -> None:
     assert module_frozen.variables == ()
     assert isinstance(module.module.module.var, tf.Variable)
     assert isinstance(module_frozen.module.module.var, tf.Tensor)
+    assert module.module.module.var.dtype is tf.float64
+    assert module_frozen.module.module.var.dtype is tf.float64
+    assert module.module.module.var == module_frozen.module.module.var
+
+
+def test_freeze_as_float32() -> None:
+    module = NestedModule(NestedModule(A()))
+    module_frozen = gpflow.utilities.freeze_as_float32(module)
+    assert len(module.variables) == 2
+    assert module_frozen.variables == ()
+    assert isinstance(module.module.module.var, tf.Variable)
+    assert isinstance(module_frozen.module.module.var, tf.Tensor)
+    assert module.module.module.var.dtype is tf.float64
+    assert module_frozen.module.module.var.dtype is tf.float32
+    assert tf.cast(module.module.module.var, tf.float32) == module_frozen.module.module.var
 
 
 def test_pickle_frozen() -> None:

gpflow-2.9.2/VERSION

@@ -1 +0,0 @@
-2.9.2