diff --git a/spinoffs/inference_gym/conftest.py b/spinoffs/inference_gym/conftest.py
index 44f0ad1d27..7dc3c71dba 100644
--- a/spinoffs/inference_gym/conftest.py
+++ b/spinoffs/inference_gym/conftest.py
@@ -15,6 +15,15 @@
 """pytest configuration."""
 
 from absl import app
+from absl import flags
+
+# TODO(siege): Find a better solution to duplicating this flag.
+flags.DEFINE_bool(
+    "use_tfds",
+    False,
+    "Whether to run tests that use TFDS.",
+    allow_override=True,
+)
 
 collect_ignore = [
     "setup.py",
@@ -26,7 +35,8 @@ def pytest_addoption(parser):
       "--absl-flag",
       action="append",
       help="flag to be passed to absl, e.g. `--absl-flag='--vary_seed'`",
-      default=[])
+      default=[],
+  )
 
 
 def pytest_collection_finish(session):
diff --git a/spinoffs/inference_gym/inference_gym/version.py b/spinoffs/inference_gym/inference_gym/version.py
index 8decc5a8d7..3a2ee1f734 100644
--- a/spinoffs/inference_gym/inference_gym/version.py
+++ b/spinoffs/inference_gym/inference_gym/version.py
@@ -17,7 +17,7 @@
 # We follow Semantic Versioning (https://semver.org/)
 _MAJOR_VERSION = '0'
 _MINOR_VERSION = '0'
-_PATCH_VERSION = '5'
+_PATCH_VERSION = '6'
 
 # When building releases, we can update this value on the release branch to
 # reflect the current release candidate ('rc0', 'rc1') or, finally, the official