Skip to content

Commit a73851a

Browse files
author
Leow, Max
committed
add exeption handling when no milestone
1 parent e2c8655 commit a73851a

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

setup.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55

66
setuptools.setup(
77
name="testrail-data",
8-
version="0.0.9",
8+
version="0.0.10",
99
install_requires=[
1010
"pandas",
1111
"testrail-api>=1.10",

testrail_data/_category.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -172,12 +172,12 @@ def get_runs(*mile_ids):
172172
if df_run2.shape[0] > 0:
173173
dataframes.append(df_run2)
174174
return dataframes
175-
176175
dfs = get_runs(milestone_ids)
177176
milestone = Milestones(self._session)
178177
df_milestone = milestone.sub_milestones_to_dataframe(*milestone_ids).filter(['id'])
179-
dfs.extend(get_runs(*df_milestone['id'].to_list()))
180-
return pd.concat(dfs).reset_index(drop=False)
178+
if df_milestone.shape[0] > 0:
179+
dfs.extend(get_runs(*df_milestone['id'].to_list()))
180+
return pd.concat(dfs).reset_index(drop=False) if dfs else pd.DataFrame()
181181

182182
def get_runs_by_plan(self, *plan_ids: int) -> list:
183183
"""

0 commit comments

Comments
 (0)