Fix to scopes

sahahn · Mar 27, 2020 · d4bccee · d4bccee
1 parent 0d17068
commit d4bccee
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/ABCD_ML/pipeline/Model_Pipeline.py b/ABCD_ML/pipeline/Model_Pipeline.py
@@ -210,8 +210,7 @@ def _get_keys_from_scope(self, scope):
 
             # Restrict all keys by the stub strs passed that arn't scopes or valid column names
             if len(restrict_keys) > 0:
-                keys += list(self.all_keys[[all([r in a for r in restrict_keys])
-                                            for a in self.all_keys]])
+                keys += [a for a in self.all_keys if all([r in a for r in restrict_keys])]
 
             # Get rid of repeats if any
             keys = list(set(keys))