Move loss to new module

invrs-io · Dec 6, 2023 · 79cc762 · 79cc762
1 parent 07c8ce1
commit 79cc762
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 3 deletions.
diff --git a/src/invrs_gym/challenges/ceviche/challenge.py b/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

diff --git a/src/invrs_gym/loss/__init__.py b/src/invrs_gym/loss/__init__.py
@@ -0,0 +1 @@
+from invrs_gym.loss import transmission_loss as transmission_loss
diff --git a/...m/challenges/ceviche/transmission_loss.py → src/invrs_gym/loss/transmission_loss.py b/...m/challenges/ceviche/transmission_loss.py → src/invrs_gym/loss/transmission_loss.py
diff --git a/...llenges/ceviche/test_transmission_loss.py → tests/loss/test_transmission_loss.py b/...llenges/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):
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1 @@
		from invrs_gym.loss import transmission_loss as transmission_loss