From 935b07d419096297ffc8b29c0eacfc985d1b6408 Mon Sep 17 00:00:00 2001
From: Klaus Greff <klausg@google.com>
Date: Wed, 23 Oct 2024 05:21:42 -0700
Subject: [PATCH] deuglify mnist example

PiperOrigin-RevId: 688922829
---
 examples/mnist_autoencoder.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/examples/mnist_autoencoder.py b/examples/mnist_autoencoder.py
index 9c1508b5..cdfbd0e2 100644
--- 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,