Update setup.py

vishwesh-vishwesh · Jun 14, 2021 · 1022473 · 1022473
1 parent 29c4500
commit 1022473
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -4,7 +4,7 @@
 
 @author: Vishwesh
 """
-import pathlib
+#import pathlib
 from setuptools import setup, find_packages
 
 #HERE = pathlib.Path(__file__).parent
@@ -35,3 +35,5 @@
       install_requires=INSTALL_REQUIRES,
       packages=find_packages()
       )
+def __getitem__(self, group) -> EntryPoints:
+        return EntryPoints(ep for ep in self if ep.group == group)