Updated comment

ajkv-google · ajkv-google · commit f59d18464f71 · 2026-01-14T00:37:26.000Z
diff --git a/recml/examples/train_hstu_jax.py b/recml/examples/train_hstu_jax.py
@@ -268,4 +268,4 @@ def main(_):
 
 
 if __name__ == "__main__":
-    app.run(main)
+    app.run(main)
diff --git a/recml/examples/train_hstu_keras.py b/recml/examples/train_hstu_keras.py
@@ -1,4 +1,4 @@
-"""HSTU Experiment Configuration using Fiddle and RecML"""
+"""HSTU Experiment Configuration using Fiddle and RecML with KerasTrainer"""
 
 import dataclasses
 from typing import Optional

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,4 @@`
`1`		`-"""HSTU Experiment Configuration using Fiddle and RecML"""`
	`1`	`+"""HSTU Experiment Configuration using Fiddle and RecML with KerasTrainer"""`
`2`	`2`
`3`	`3`	`import dataclasses`
`4`	`4`	`from typing import Optional`