mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
b6bdb67f82
Changes by apply order:
1. Replace all `".."` and `os.pardir` usage with `os.path.dirname(...)`.
2. Replace nested `os.path.dirname(os.path.dirname(...))` call with `str(Path(...).parent.parent)`.
3. Reorder `.absolute()` ~/ `.resolve()`~ and `.parent`: always resolve the path first.
`.parent{...}.absolute()` -> `.absolute().parent{...}`
4. Replace chained `.parent x N` with `.parents[${N - 1}]`: the code is easier to read (see 5.)
`.parent.parent.parent.parent` -> `.parents[3]`
5. ~Replace `.parents[${N - 1}]` with `.parents[${N} - 1]`: the code is easier to read and does not introduce any runtime overhead.~
~`.parents[3]` -> `.parents[4 - 1]`~
6. ~Replace `.parents[2 - 1]` with `.parent.parent`: because the code is shorter and easier to read.~
Pull Request resolved: https://github.com/pytorch/pytorch/pull/129374
Approved by: https://github.com/justinchuby, https://github.com/malfet
27 lines
518 B
Python
Executable File
27 lines
518 B
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
import os
|
|
import sys
|
|
|
|
import yaml
|
|
|
|
|
|
# Need to import modules that lie on an upward-relative path
|
|
sys.path.append(os.path.dirname(sys.path[0]))
|
|
|
|
import cimodel.lib.miniyaml as miniyaml
|
|
|
|
|
|
def regurgitate(depth, use_pyyaml_formatter=False):
|
|
data = yaml.safe_load(sys.stdin)
|
|
|
|
if use_pyyaml_formatter:
|
|
output = yaml.dump(data, sort_keys=True)
|
|
sys.stdout.write(output)
|
|
else:
|
|
miniyaml.render(sys.stdout, data, depth)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
regurgitate(3)
|