Skip to content

Commit d7c9f67

Browse files
committed
fix(Parser): exception
1 parent e991f7e commit d7c9f67

File tree

2 files changed

+6
-8
lines changed

2 files changed

+6
-8
lines changed

rrparser/Parser.py

Lines changed: 4 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -15,9 +15,6 @@
1515
read_csv,
1616
DataFrame
1717
)
18-
# from pandas.errors import (
19-
# UndefinedVariableError
20-
# )
2118
from csv import (
2219
QUOTE_ALL,
2320
QUOTE_NONE
@@ -147,10 +144,10 @@ def filter_(
147144
query += ' & `Rule usage` == @rule_usage_filter'
148145
# Used in 'e' in case of exception raised
149146
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)
154151

155152

156153
def fetch_retrorules(

tests/test_light.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -69,7 +69,8 @@ def test_BadInputFormatCSV_1(self):
6969
rules_file = self.rules_file,
7070
input_format = 'tsv',
7171
diameters = diam,
72-
outfile = outfile.name)
72+
outfile = outfile.name
73+
)
7374
outfile.close()
7475

7576

0 commit comments

Comments
 (0)