invrs-opt 0.3.2__tar.gz → 0.4.0__tar.gz

{invrs_opt-0.3.2/src/invrs_opt.egg-info → invrs_opt-0.4.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: invrs_opt
-Version: 0.3.2
+Version: 0.4.0
 Summary: Algorithms for inverse design
 Author-email: "Martin F. Schubert" <mfschubert@gmail.com>
 Maintainer-email: "Martin F. Schubert" <mfschubert@gmail.com>
@@ -33,8 +33,10 @@ License-File: LICENSE
 Requires-Dist: jax
 Requires-Dist: jaxlib
 Requires-Dist: numpy
+Requires-Dist: requests
 Requires-Dist: scipy
 Requires-Dist: totypes
+Requires-Dist: types-requests
 Provides-Extra: tests
 Requires-Dist: parameterized; extra == "tests"
 Requires-Dist: pytest; extra == "tests"
@@ -47,7 +49,7 @@ Requires-Dist: mypy; extra == "dev"
 Requires-Dist: pre-commit; extra == "dev"
 
 # invrs-opt - Optimization algorithms for inverse design
-`v0.3.2`
+`v0.4.0`
 
 ## Overview
 

{invrs_opt-0.3.2 → invrs_opt-0.4.0}/README.md

@@ -1,5 +1,5 @@
 # invrs-opt - Optimization algorithms for inverse design
-`v0.3.2`
+`v0.4.0`
 
 ## Overview
 

{invrs_opt-0.3.2 → invrs_opt-0.4.0}/pyproject.toml

@@ -1,7 +1,7 @@
 [project]
 
 name = "invrs_opt"
-version = "v0.3.2"
+version = "v0.4.0"
 description = "Algorithms for inverse design"
 keywords = ["topology", "optimization", "jax", "inverse design"]
 readme = "README.md"
@@ -19,8 +19,10 @@ dependencies = [
     "jax",
     "jaxlib",
     "numpy",
+    "requests",
     "scipy",
     "totypes",
+    "types-requests",
 ]
 
 [project.optional-dependencies]

{invrs_opt-0.3.2 → invrs_opt-0.4.0}/src/invrs_opt/__init__.py

@@ -3,7 +3,7 @@
 Copyright (c) 2023 The INVRS-IO authors.
 """
 
-__version__ = "v0.3.2"
+__version__ = "v0.4.0"
 __author__ = "Martin F. Schubert <mfschubert@gmail.com>"
 
 from invrs_opt.lbfgsb.lbfgsb import density_lbfgsb as density_lbfgsb

invrs_opt-0.4.0/src/invrs_opt/experimental/client.py

@@ -0,0 +1,152 @@
+"""Defines basic client optimizers for use with an optimization service.
+
+Copyright (c) 2023 The INVRS-IO authors.
+"""
+
+import json
+import requests
+import time
+from typing import Any, Dict, Optional
+
+from totypes import json_utils
+
+from invrs_opt import base
+from invrs_opt.experimental import labels
+
+
+PyTree = Any
+StateToken = str
+
+SESSION = None
+SERVER_ADDRESS = None
+
+
+def login(server_address: str) -> None:
+    """Set the global server address and create a requests session."""
+    global SESSION
+    global SERVER_ADDRESS
+    SESSION = requests.Session()
+    SERVER_ADDRESS = server_address
+
+
+def optimizer_client(
+    algorithm: str,
+    hparams: Dict[str, Any],
+    server_address: Optional[str],
+    session: Optional[requests.Session],
+) -> base.Optimizer:
+    """Generic optimizer class."""
+
+    if server_address is None:
+        if SERVER_ADDRESS is None:
+            raise ValueError(
+                "Argument `server_address` and the global `SERVER_ADDRESS` cannot "
+                "both be `None`. Use the `login` method to set the global, or "
+                "explicitly provide a value."
+            )
+    if session is None:
+        if SESSION is None:
+            raise ValueError(
+                "Argument `session` and the global `SESSION` cannot "
+                "both be `None`. Use the `login` method to set the global, or "
+                "explicitly provide a value."
+            )
+        session = SESSION
+
+    opt_config = {
+        labels.ALGORITHM: algorithm,
+        labels.HPARAMS: hparams,
+    }
+
+    def init_fn(params: PyTree) -> StateToken:
+        """Handles 'init' requests."""
+        serialized_data = json_utils.json_from_pytree(
+            dict(opt_config=opt_config, data={"params": params})
+        )
+        post_response = session.post(
+            f"{SERVER_ADDRESS}/{labels.ROUTE_INIT}/", data=serialized_data
+        )
+
+        if not post_response.status_code == 200:
+            raise requests.RequestException(post_response.text)
+        response = json.loads(post_response.text)
+        new_state_token: str = response[labels.STATE_TOKEN]
+        return new_state_token
+
+    def update_fn(
+        *,
+        grad: PyTree,
+        value: float,
+        params: PyTree,
+        state: StateToken,
+    ) -> StateToken:
+        """Handles 'update' requests."""
+        state_token = state
+        del state
+        serialized_data = json_utils.json_from_pytree(
+            {
+                labels.OPT_CONFIG: opt_config,
+                labels.DATA: {
+                    labels.PARAMS: params,
+                    labels.VALUE: value,
+                    labels.GRAD: grad,
+                    labels.STATE_TOKEN: state_token,
+                },
+            }
+        )
+        post_response = session.post(
+            f"{SERVER_ADDRESS}/{labels.ROUTE_UPDATE}/{state_token}/",
+            data=serialized_data,
+        )
+
+        if not post_response.status_code == 200:
+            raise requests.RequestException(post_response.text)
+        response = json.loads(post_response.text)
+        new_state_token: str = response[labels.STATE_TOKEN]
+        return new_state_token
+
+    def params_fn(
+        state: StateToken,
+        timeout: float = 60.0,
+        poll_interval: float = 0.1,
+    ) -> PyTree:
+        """Handles 'params' requests."""
+        state_token = state
+        del state
+        assert timeout >= poll_interval
+        start_time = time.time()
+        while time.time() < start_time + timeout:
+            get_response = session.get(
+                f"{SERVER_ADDRESS}/{labels.ROUTE_PARAMS}/{state_token}"
+            )
+            if get_response.status_code == 200:
+                break
+            elif get_response.status_code == 404 and get_response.text.endswith(
+                labels.MESSAGE_STATE_NOT_READY.format(state_token)
+            ):
+                time.sleep(poll_interval)
+            else:
+                raise requests.RequestException(get_response.text)
+
+        if not get_response.status_code == 200:
+            raise requests.Timeout("Timed out while waiting for params.")
+        response = json.loads(get_response.text)
+        return json_utils.pytree_from_json(response[labels.PARAMS])
+
+    return base.Optimizer(init=init_fn, update=update_fn, params=params_fn)
+
+
+# -----------------------------------------------------------------------------
+# Specific optimizers implemented here.
+# -----------------------------------------------------------------------------
+
+
+def lbfgsb(maxcor: int = 20, line_search_max_steps: int = 100) -> base.Optimizer:
+    """Optimizer implementing the L-BFGS-B scheme."""
+    hparams = {
+        "maxcor": maxcor,
+        "line_search_max_steps": line_search_max_steps,
+    }
+    return optimizer_client(
+        algorithm="lbfgsb", hparams=hparams, server_address=None, session=None
+    )

invrs_opt-0.4.0/src/invrs_opt/experimental/labels.py

@@ -0,0 +1,23 @@
+"""Defines labels and messages used in the context of an optimization service.
+
+Copyright (c) 2023 The INVRS-IO authors.
+"""
+
+VALUE = "value"
+GRAD = "grad"
+PARAMS = "params"
+STATE_TOKEN = "state_token"
+MESSAGE = "message"
+
+OPT_CONFIG = "opt_config"
+ALGORITHM = "algorithm"
+HPARAMS = "hparams"
+DATA = "data"
+
+ROUTE_INIT = "init"
+ROUTE_UPDATE = "update"
+ROUTE_PARAMS = "params"
+
+MESSAGE_STATE_NOT_KNOWN = "State token {} was not recognized."
+MESSAGE_STATE_NOT_READY = "State for token {} is not ready."
+MESSAGE_STATE_NOT_VALID = "State for token {} is not valid."

{invrs_opt-0.3.2 → invrs_opt-0.4.0}/src/invrs_opt/lbfgsb/lbfgsb.py

@@ -261,10 +261,12 @@ def transformed_lbfgsb(
             flat_latent_params = jnp.asarray(scipy_lbfgsb_state.x)
             return flat_latent_params, scipy_lbfgsb_state.to_jax()
 
-        params, latent_params, jax_lbfgsb_state = state
+        _, latent_params, jax_lbfgsb_state = state
         _, vjp_fn = jax.vjp(transform_fn, latent_params)
         (latent_grad,) = vjp_fn(grad)
-        flat_latent_grad, unflatten_fn = flatten_util.ravel_pytree(latent_grad)
+        flat_latent_grad, unflatten_fn = flatten_util.ravel_pytree(
+            latent_grad
+        )  # type: ignore[no-untyped-call]
 
         (
             flat_latent_params,

{invrs_opt-0.3.2 → invrs_opt-0.4.0/src/invrs_opt.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: invrs_opt
-Version: 0.3.2
+Version: 0.4.0
 Summary: Algorithms for inverse design
 Author-email: "Martin F. Schubert" <mfschubert@gmail.com>
 Maintainer-email: "Martin F. Schubert" <mfschubert@gmail.com>
@@ -33,8 +33,10 @@ License-File: LICENSE
 Requires-Dist: jax
 Requires-Dist: jaxlib
 Requires-Dist: numpy
+Requires-Dist: requests
 Requires-Dist: scipy
 Requires-Dist: totypes
+Requires-Dist: types-requests
 Provides-Extra: tests
 Requires-Dist: parameterized; extra == "tests"
 Requires-Dist: pytest; extra == "tests"
@@ -47,7 +49,7 @@ Requires-Dist: mypy; extra == "dev"
 Requires-Dist: pre-commit; extra == "dev"
 
 # invrs-opt - Optimization algorithms for inverse design
-`v0.3.2`
+`v0.4.0`
 
 ## Overview
 

{invrs_opt-0.3.2 → invrs_opt-0.4.0}/src/invrs_opt.egg-info/SOURCES.txt

@@ -9,6 +9,9 @@ src/invrs_opt.egg-info/SOURCES.txt
 src/invrs_opt.egg-info/dependency_links.txt
 src/invrs_opt.egg-info/requires.txt
 src/invrs_opt.egg-info/top_level.txt
+src/invrs_opt/experimental/__init__.py
+src/invrs_opt/experimental/client.py
+src/invrs_opt/experimental/labels.py
 src/invrs_opt/lbfgsb/__init__.py
 src/invrs_opt/lbfgsb/lbfgsb.py
 src/invrs_opt/lbfgsb/transform.py

{invrs_opt-0.3.2 → invrs_opt-0.4.0}/src/invrs_opt.egg-info/requires.txt

@@ -1,8 +1,10 @@
 jax
 jaxlib
 numpy
+requests
 scipy
 totypes
+types-requests
 
 [dev]
 bump-my-version