From 10224731d7c5057e6561e04709d7223e78619d55 Mon Sep 17 00:00:00 2001
From: Vishwesh <36016864+vishwesh-vishwesh@users.noreply.github.com>
Date: Sun, 13 Jun 2021 21:56:03 -0700
Subject: [PATCH] Update setup.py

---
 setup.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 585dcc6..b489307 100644
--- 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)