Function load_named_sharding_json

tests/sharding_dump.py:251–255

Loads the named_shardings.json file into a plain Python dict.

(json_path: str | Path)

def load_named_sharding_json(json_path: str | Path) -> dict:
  """Loads the named_shardings.json file into a plain Python dict."""
  json_path = Path(json_path)
  with open(json_path, "r", encoding="utf-8") as f:
    return json.load(f)


def main(argv: Sequence[str]) -> None: