diff --git a/Otherfiles/requirements-splitter.py b/Otherfiles/requirements-splitter.py index 18c8108c..e3f563a4 100644 --- a/Otherfiles/requirements-splitter.py +++ b/Otherfiles/requirements-splitter.py @@ -7,11 +7,13 @@ with open('dev-requirements.txt', 'r') as f: for line in f: + plot_lib_flag = False for lib in PLOT_LIB_LIST: if line.find(lib) != -1: + plot_lib_flag = True plot_req += line break - if '==' not in line: + if '==' not in line and not plot_lib_flag: test_req += line with open('test-requirements.txt', 'w') as f: