def __init__(self, assets_dir=None, requirements_file="requirements.yml"):
        self.assets_dir = assets_dir or Path(".") / "assets"
        self.requirements = GroupedRequirements()
        self.requirements.load(requirements_file)
        self.temp_dir = mkdtemp()
def _stats_to_str(self, stats: Dict[str, str]) -> str:
        stage = f"{self._stage.upper()} " if self._stage is not None else ""
        output = [stage + "Profiler Report"]
        for action, value in stats.items():
            header = f"Profile stats for: {action}"
            if self._local_rank is not None:
                header += f" rank: {self._local_rank}"
            output.append(header)
            output.append(value)
        return os.linesep.join(output)