update requirements

erdogant · erdogant · commit b89552b17d60 · 2024-10-11T13:42:34.000+02:00
diff --git a/requirements.txt b/requirements.txt
@@ -0,0 +1,6 @@
+scikit-learn
+numpy
+graphviz>=0.20.1
+matplotlib
+funcsigs
+setgraphviz
diff --git a/setup.py b/setup.py
@@ -13,7 +13,7 @@
 with open("README.md", "r", encoding='utf8') as fh:
     long_description = fh.read()
 setuptools.setup(
-     install_requires=['scikit-learn','numpy','graphviz>=0.20.1','matplotlib', 'funcsigs'],
+     install_requires=['scikit-learn','numpy','graphviz>=0.20.1','matplotlib', 'funcsigs', 'setgraphviz'],
      python_requires='>=3',
      name='treeplot',
      version=new_version,
