Resolve Linux Wheels and Pyre

aporialiao · facebook-github-bot · commit d26b1c49211f · 2025-04-11T10:24:30.000-07:00
Summary:
```
ƛ Found 1 type error!

torchrec/distributed/planner/stats.py:1111:64 Incompatible parameter type [6]: In call `pd.core.frame.DataFrame.from_dict`, for argument `columns`, expected `Union[None, ndarray[typing.Any, typing.Any], SequenceNotStr[typing.Any], ExtensionArray, Index, Series, range]` but got `List[str]`.

```
Resolving this error from D72410003 for Github pyre ci

Differential Revision: D72871677
diff --git 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
@@ -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,

Original file line number	Diff line number	Diff line change
`@@ -29,3 +29,4 @@ fi`
`29`	`29`
`30`	`30`
`31`	`31`	`${CONDA_RUN} pip install importlib-metadata`
	`32`	`+${CONDA_RUN} pip install pandas`
Original file line number	Diff line number	Diff line change
`@@ -1108,6 +1108,7 @@ def _calculate_critical_path(best_plan: List[ShardingOption]) -> CriticalPathEst`
`1108`	`1108`	`perf_type.split("_")[1], # compute or comms`
`1109`	`1109`	`)`
`1110`	`1110`	`] += perf_breakdown[perf_type]`
	`1111`	`+ # pyre-ignore[6]`
`1111`	`1112`	`perf_df = pd.DataFrame.from_dict(perf_data, orient="index", columns=["perf"])`
`1112`	`1113`	`perf_df.index = pd.MultiIndex.from_tuples(`
`1113`	`1114`	`perf_df.index,`