diff --git a/.github/scripts/install_libs.sh b/.github/scripts/install_libs.sh index 27522ff92..c7b7d2938 100644 --- a/.github/scripts/install_libs.sh +++ b/.github/scripts/install_libs.sh @@ -29,3 +29,4 @@ fi ${CONDA_RUN} pip install importlib-metadata +${CONDA_RUN} pip install pandas diff --git a/torchrec/distributed/planner/stats.py b/torchrec/distributed/planner/stats.py index df548d489..c119e4424 100644 --- a/torchrec/distributed/planner/stats.py +++ b/torchrec/distributed/planner/stats.py @@ -1108,6 +1108,7 @@ def _calculate_critical_path(best_plan: List[ShardingOption]) -> CriticalPathEst perf_type.split("_")[1], # compute or comms ) ] += perf_breakdown[perf_type] + # pyre-ignore[6] perf_df = pd.DataFrame.from_dict(perf_data, orient="index", columns=["perf"]) perf_df.index = pd.MultiIndex.from_tuples( perf_df.index,