diff --git a/sdks/python/tox.ini b/sdks/python/tox.ini
index ac583f3b6c79..882125f0f701 100644
--- a/sdks/python/tox.ini
+++ b/sdks/python/tox.ini
@@ -187,7 +187,7 @@ setenv =
   TC_SLEEP_TIME = {env:TC_SLEEP_TIME:1}
 
 # NOTE: we could add ml_test to increase the collected code coverage metrics, but it would make the suite slower.
-extras = test,hadoop,gcp,interactive,dataframe,aws,redis
+extras = test,hadoop,gcp,interactive,dataframe,aws,redis,tfrecord,yaml
 commands =
   bash {toxinidir}/scripts/run_pytest.sh {envname} "{posargs}" "--cov-report=xml --cov=. --cov-append"