From 3b9ae56b6bc4048c7791bdd8a6eee00995f9a606 Mon Sep 17 00:00:00 2001
From: Scott Sievert <stsievert@users.noreply.github.com>
Date: Mon, 28 Aug 2017 15:17:54 -0500
Subject: [PATCH] Clean __contains__

---
 pyradigm/pyradigm.py | 7 ++-----
 1 file changed, 2 insertions(+), 5 deletions(-)

diff --git a/pyradigm/pyradigm.py b/pyradigm/pyradigm.py
index 2738bc2..4fd7418 100644
--- a/pyradigm/pyradigm.py
+++ b/pyradigm/pyradigm.py
@@ -415,11 +415,8 @@ def get_subset(self, subset_ids):
             return MLDataset()
 
     def __contains__(self, item):
-        "Boolean test of membership of a sample in the dataset."
-        if item in self.keys:
-            return True
-        else:
-            return False
+        """Boolean test of membership of a sample in the dataset."""
+        return item in self.keys
 
     @staticmethod
     def __get_subset_from_dict(input_dict, subset):