Skip to content
Snippets Groups Projects
Commit adf17fb2 authored by JasonJiazhiZhang's avatar JasonJiazhiZhang Committed by Oleksandr
Browse files

Limit requirements search scope to habitat_baselines folder (#132)

Limit requirements search scope to habitat_baselines folder.
parent ec9557a3
No related branches found
No related tags found
No related merge requests found
......@@ -35,7 +35,9 @@ BASELINE_PATH = ["habitat_baselines", "habitat_baselines.*"]
DEFAULT_EXCLUSION = ["test", "examples"]
FULL_REQUIREMENTS = set()
# collect requirements.txt file in all subdirectories
for file_name in glob.glob("**/requirements.txt", recursive=True):
for file_name in ["requirements.txt"] + glob.glob(
"habitat_baselines/**/requirements.txt", recursive=True
):
with open(file_name) as f:
reqs = f.read()
FULL_REQUIREMENTS.update(reqs.strip().split("\n"))
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment