replace class name of EvaporationPrecipitationDiagnostics with Precip…

…itationDiagnosticsConstrained PiperOrigin-RevId: 676191337
google-research · Sep 19, 2024 · 4489014 · 4489014
1 parent 9d89731
commit 4489014
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/neuralgcm/diagnostics.py b/neuralgcm/diagnostics.py
@@ -264,7 +264,7 @@ def __call__(
 
 # TODO(janniyuval) add a decoder that can add some Gaussian noise to evap/precip
 @gin.register
-class EvaporationPrecipitationDiagnostics(
+class PrecipitationDiagnosticsConstrained(
     hk.Module, PrecipitationMinusEvaporationDiagnostics
 ):
   """Predict evaporation and computes cumulative precipitation.