- env_files = [f for f in files
- if path.splitext(f)[-1] == ".env"]
- yaml_files = [f for f in files
- if path.splitext(f)[-1] in ['.yml', '.yaml']]
+ env_files = [f for f in files if path.splitext(f)[-1] == ".env"]
+ yaml_files = [f for f in files if path.splitext(f)[-1] in [".yml", ".yaml"]]