diff --git a/sdk/python/requirements/py3.10-ci-requirements.txt b/sdk/python/requirements/py3.10-ci-requirements.txt
index 523b71874b..8c2c8565df 100644
--- a/sdk/python/requirements/py3.10-ci-requirements.txt
+++ b/sdk/python/requirements/py3.10-ci-requirements.txt
@@ -159,6 +159,7 @@ distlib==0.3.8
 docker==7.0.0
     # via
     #   feast (setup.py)
+    #   moto
     #   testcontainers
 docutils==0.19
     # via sphinx
@@ -450,7 +451,7 @@ mock==2.0.0
     # via feast (setup.py)
 moreorless==0.4.0
     # via bowler
-moto==5.0.0
+moto[dynamodb]==5.0.0
     # via feast (setup.py)
 msal==1.26.0
     # via
@@ -613,6 +614,8 @@ py==1.11.0
     # via feast (setup.py)
 py-cpuinfo==9.0.0
     # via pytest-benchmark
+py-partiql-parser==0.5.0
+    # via moto
 py4j==0.10.9.7
     # via pyspark
 pyarrow==15.0.0
diff --git a/setup.py b/setup.py
index c9d2bcd944..ef52534202 100644
--- a/setup.py
+++ b/setup.py
@@ -155,7 +155,7 @@
         "grpcio-testing>=1.56.2,<2",
         "minio==7.1.0",
         "mock==2.0.0",
-        "moto",
+        "moto[dynamodb]",
         "mypy>=0.981,<0.990",
         "avro==1.10.0",
         "fsspec<2023.10.0",