1 files changed, 4 insertions, 1 deletions
diff --git a/setup.py b/setup.py
index fbf834f..2eb7009 100644
--- a/setup.py
+++ b/setup.py
@@ -45,5 +45,8 @@ setup(
         "Programming Language :: Python :: 3.7",
         "Programming Language :: Python :: 3.8",
     ],
-    extras_require={"autocompletion": ["argcomplete>=1.10.0,<2"]},
+    extras_require={
+        "autocompletion": ["argcomplete>=1.10.0,<2"],
+        "yaml": ["PyYaml>=5.2"],
+    },
 )