Fix ceviche loss hyperparameters

.bumpversion.toml

@@ -1,5 +1,5 @@
 [tool.bumpversion]
-current_version = "v0.3.1"
+current_version = "v0.3.2"
 commit = true
 commit_args = "--no-verify"
 tag = true

README.md

@@ -1,5 +1,5 @@
 # invrs-gym - A collection of inverse design challenges
-`v0.3.1`
+`v0.3.2`
 
 ## Overview
 The `invrs_gym` package is an open-source gym containing a diverse set of photonic design challenges, which are relevant for a wide range of applications such as AR/VR, optical networking, LIDAR, and others.

pyproject.toml

@@ -1,7 +1,7 @@
 [project]
 
 name = "invrs_gym"
-version = "v0.3.1"
+version = "v0.3.2"
 description = "A collection of inverse design challenges"
 keywords = ["topology", "optimization", "jax", "inverse design"]
 readme = "README.md"

src/invrs_gym/__init__.py

@@ -3,7 +3,7 @@
 Copyright (c) 2023 The INVRS-IO authors.
 """
 
-__version__ = "v0.3.1"
+__version__ = "v0.3.2"
 __author__ = "Martin F. Schubert <[email protected]>"
 
 from invrs_gym import challenges as challenges

src/invrs_gym/challenges/ceviche/challenge.py

@@ -19,14 +19,15 @@
 
 from invrs_gym import utils
 from invrs_gym.challenges import base
-from invrs_gym.challenges.ceviche import defaults, transmission_loss
+from invrs_gym.challenges.ceviche import defaults
+from invrs_gym.loss import transmission_loss
 
 Params = Any
 
 FIELDS = "fields"
 SPARAMS = "sparams"
 
-TRANSMISSION_EXPONENT = 0.5
+TRANSMISSION_EXPONENT = 1.0
 SCALAR_EXPONENT = 2.0
 
 DENSITY_LOWER_BOUND = 0.0

src/invrs_gym/challenges/sorter/common.py

@@ -304,10 +304,9 @@ def simulate_sorter(
 
     This code is adapted from the fmmax.examples.sorter script.
 
-    The sorter consists of a metasurface layer situated above a quad of pixels.
-    Each pixel is square in shape, and includes a circular "target" region in
-    its interior. Above the metasurface is a cap, and it is separated from the
-    substrate by a spacer layer, as illustrated below.
+    The sorter consists of a metasurface layer situated above a quad of pixels,
+    with each pixel being square in shape. Above the metasurface is a cap, and
+    it is separated from the substrate by a spacer layer as illustrated below.
 
                                __________________________
                               /                         /|

src/invrs_gym/loss/__init__.py

@@ -0,0 +1 @@
+from invrs_gym.loss import transmission_loss as transmission_loss

tests/challenges/ceviche/test_transmission_loss.py → tests/loss/test_transmission_loss.py

@@ -6,7 +6,7 @@
 import functools
 import unittest
 
-from invrs_gym.challenges.ceviche import transmission_loss
+from invrs_gym.loss import transmission_loss
 
 
 class OrthotopeSmoothLossTest(unittest.TestCase):