File tree 2 files changed +6
-8
lines changed
2 files changed +6
-8
lines changed Original file line number Diff line number Diff line change 15
15
read_csv ,
16
16
DataFrame
17
17
)
18
- # from pandas.errors import (
19
- # UndefinedVariableError
20
- # )
21
18
from csv import (
22
19
QUOTE_ALL ,
23
20
QUOTE_NONE
@@ -147,10 +144,10 @@ def filter_(
147
144
query += ' & `Rule usage` == @rule_usage_filter'
148
145
# Used in 'e' in case of exception raised
149
146
rule_usage_filter = ['both' , rule_type ]
150
- # try:
151
- return df .query (query )
152
- # except UndefinedVariableError as e:
153
- # raise KeyError(e)
147
+ try :
148
+ return df .query (query )
149
+ except Exception as e :
150
+ raise KeyError (e )
154
151
155
152
156
153
def fetch_retrorules (
Original file line number Diff line number Diff line change @@ -69,7 +69,8 @@ def test_BadInputFormatCSV_1(self):
69
69
rules_file = self .rules_file ,
70
70
input_format = 'tsv' ,
71
71
diameters = diam ,
72
- outfile = outfile .name )
72
+ outfile = outfile .name
73
+ )
73
74
outfile .close ()
74
75
75
76
You can’t perform that action at this time.
0 commit comments