deuglify mnist example

PiperOrigin-RevId: 688922829
google-research · Oct 23, 2024 · 935b07d · 935b07d
1 parent ebb9144
commit 935b07d
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/examples/mnist_autoencoder.py b/examples/mnist_autoencoder.py
@@ -87,8 +87,7 @@ def get_config():
 
 
 def _make_ds(training: bool):
-  Tfds = kd.data.py.Tfds
-  return Tfds(
+  return kd.data.py.Tfds(
       name="mnist",
       split="train" if training else "test",
       shuffle=True if training else False,