fix ci build

Author: rafaelherik

setup.py

@@ -20,6 +20,7 @@
     },
     python_requires='>=3.10',
     install_requires=[
+        'pytest>=7.0.0',
     ],
     classifiers=[
         'Development Status :: 3 - Alpha',

tfsumpy/plan_analyzer.py

@@ -1,6 +1,5 @@
 import re
 import json
-import yaml
 import logging
 import os
 from typing import Dict, List